Split gvcf2numpy command into import and export-numpy.

[lightning.git] / exportnumpy_test.go

diff --git a/gvcf2numpy_test.go b/exportnumpy_test.go
similarity index 62%
rename from gvcf2numpy_test.go
rename to exportnumpy_test.go
index 21e08f9ac5cd58d0451d09a9e6cc588acf3d1226..4bef7bd1e88ec7147d7b8382b997f83404b5f760 100644 (file)
--- a/gvcf2numpy_test.go
+++ b/exportnumpy_test.go
@@ -8,16 +8,18 @@ import (
        "gopkg.in/check.v1"
 )
 
-type gvcf2numpySuite struct{}
+type exportSuite struct{}
 
-var _ = check.Suite(&gvcf2numpySuite{})
+var _ = check.Suite(&exportSuite{})
 
-func (s *gvcf2numpySuite) TestFastaToNumpy(c *check.C) {
-       var stdout bytes.Buffer
-       var cmd gvcf2numpy
-       exited := cmd.RunCommand("gvcf2numpy", []string{"-tag-library", "testdata/tags", "-ref", "testdata/ref", "testdata/a.1.fasta"}, &bytes.Buffer{}, &stdout, os.Stderr)
+func (s *exportSuite) TestFastaToNumpy(c *check.C) {
+       var buffer bytes.Buffer
+       exited := (&importer{}).RunCommand("import", []string{"-tag-library", "testdata/tags", "-ref", "testdata/ref", "testdata/a.1.fasta"}, &bytes.Buffer{}, &buffer, os.Stderr)
        c.Check(exited, check.Equals, 0)
-       npy, err := gonpy.NewReader(&stdout)
+       var output bytes.Buffer
+       exited = (&exportNumpy{}).RunCommand("export-numpy", nil, &buffer, &output, os.Stderr)
+       c.Check(exited, check.Equals, 0)
+       npy, err := gonpy.NewReader(&output)
        c.Assert(err, check.IsNil)
        variants, err := npy.GetUint16()
        c.Assert(err, check.IsNil)