Reduce import RAM.

[lightning.git] / exportnumpy_test.go

diff --git a/exportnumpy_test.go b/exportnumpy_test.go
index ac0e552a035aacbdd990b3bb644877518c420b86..9dd71da82b833014ef826cbad6c9e0db05988e87 100644 (file)
--- a/exportnumpy_test.go
+++ b/exportnumpy_test.go
@@ -1,4 +1,4 @@
-package main
+package lightning
 
 import (
        "bytes"
@@ -21,21 +21,21 @@ func (s *exportSuite) TestFastaToNumpy(c *check.C) {
        c.Check(exited, check.Equals, 0)
        npy, err := gonpy.NewReader(&output)
        c.Assert(err, check.IsNil)
-       variants, err := npy.GetUint16()
+       variants, err := npy.GetInt16()
        c.Assert(err, check.IsNil)
        for i := 0; i < 4; i += 2 {
                if variants[i] == 1 {
-                       c.Check(variants[i+1], check.Equals, uint16(2), check.Commentf("i=%d, v=%v", i, variants))
+                       c.Check(variants[i+1], check.Equals, int16(2), check.Commentf("i=%d, v=%v", i, variants))
                } else {
-                       c.Check(variants[i], check.Equals, uint16(2), check.Commentf("i=%d, v=%v", i, variants))
+                       c.Check(variants[i], check.Equals, int16(2), check.Commentf("i=%d, v=%v", i, variants))
                }
        }
        for i := 4; i < 9; i += 2 {
-               c.Check(variants[i], check.Equals, uint16(1), check.Commentf("i=%d, v=%v", i, variants))
+               c.Check(variants[i], check.Equals, int16(1), check.Commentf("i=%d, v=%v", i, variants))
        }
 }
 
-func sortUints(variants []uint16) {
+func sortUints(variants []int16) {
        for i := 0; i < len(variants); i += 2 {
                if variants[i] > variants[i+1] {
                        for j := 0; j < len(variants); j++ {
@@ -49,20 +49,20 @@ func sortUints(variants []uint16) {
 func (s *exportSuite) TestOnehot(c *check.C) {
        for _, trial := range []struct {
                incols  int
-               in      []uint16
+               in      []int16
                outcols int
-               out     []uint16
+               out     []int16
        }{
-               {2, []uint16{1, 1, 1, 1}, 2, []uint16{1, 1, 1, 1}},
-               {2, []uint16{1, 1, 1, 2}, 3, []uint16{1, 1, 0, 1, 0, 1}},
+               {2, []int16{1, 1, 1, 1}, 2, []int16{1, 1, 1, 1}},
+               {2, []int16{1, 1, 1, 2}, 3, []int16{1, 1, 0, 1, 0, 1}},
                {
                        // 2nd column => 3 one-hot columns
                        // 4th column => 0 one-hot columns
-                       4, []uint16{
+                       4, []int16{
                                1, 1, 0, 0,
                                1, 2, 1, 0,
                                1, 3, 0, 0,
-                       }, 5, []uint16{
+                       }, 5, []int16{
                                1, 1, 0, 0, 0,
                                1, 0, 1, 0, 1,
                                1, 0, 0, 1, 0,