Fix reference assembly.

[lightning.git] / slice_test.go

diff --git a/slice_test.go b/slice_test.go
index 0d1fe3a4c6a92d3115c059ccff07cf7fbee5b4b1..e59821480ecc19023ad35888d013b1aed5fa9aca 100644 (file)
--- a/slice_test.go
+++ b/slice_test.go
@@ -9,6 +9,7 @@ import (
        "os"
        "os/exec"
 
+       "github.com/kshedden/gonpy"
        "gopkg.in/check.v1"
 )
 
@@ -22,6 +23,7 @@ func (s *sliceSuite) TestImportAndSlice(c *check.C) {
        err := ioutil.WriteFile(tmpdir+"/chr1-12-100.bed", []byte("chr1\t12\t100\ttest.1\n"), 0644)
        c.Check(err, check.IsNil)
 
+       c.Log("=== import testdata/ref ===")
        exited := (&importer{}).RunCommand("import", []string{
                "-local=true",
                "-tag-library", "testdata/tags",
@@ -32,16 +34,17 @@ func (s *sliceSuite) TestImportAndSlice(c *check.C) {
        }, nil, os.Stderr, os.Stderr)
        c.Assert(exited, check.Equals, 0)
 
+       c.Log("=== import testdata/pipeline1 ===")
        exited = (&importer{}).RunCommand("import", []string{
                "-local=true",
                "-tag-library", "testdata/tags",
                "-output-tiles",
-               // "-save-incomplete-tiles",
                "-o", tmpdir + "/library2.gob",
                "testdata/pipeline1",
        }, nil, os.Stderr, os.Stderr)
        c.Assert(exited, check.Equals, 0)
 
+       c.Log("=== merge ===")
        exited = (&merger{}).RunCommand("merge", []string{
                "-local=true",
                "-o", tmpdir + "/library.gob",
@@ -51,14 +54,50 @@ func (s *sliceSuite) TestImportAndSlice(c *check.C) {
        c.Assert(exited, check.Equals, 0)
 
        input := tmpdir + "/library.gob"
+       slicedir := c.MkDir()
 
+       c.Log("=== slice ===")
        exited = (&slicecmd{}).RunCommand("slice", []string{
                "-local=true",
                "-input-dir=" + input,
-               "-output-dir=" + tmpdir,
+               "-output-dir=" + slicedir,
                "-tags-per-file=2",
        }, nil, os.Stderr, os.Stderr)
        c.Check(exited, check.Equals, 0)
-       out, _ := exec.Command("find", tmpdir, "-ls").CombinedOutput()
+       out, _ := exec.Command("find", slicedir, "-ls").CombinedOutput()
        c.Logf("%s", out)
+
+       c.Log("=== slice-numpy ===")
+       npydir := c.MkDir()
+       exited = (&sliceNumpy{}).RunCommand("slice-numpy", []string{
+               "-local=true",
+               "-input-dir=" + slicedir,
+               "-output-dir=" + npydir,
+       }, nil, os.Stderr, os.Stderr)
+       c.Check(exited, check.Equals, 0)
+       out, _ = exec.Command("find", npydir, "-ls").CombinedOutput()
+       c.Logf("%s", out)
+
+       f, err := os.Open(npydir + "/matrix.0000.npy")
+       c.Assert(err, check.IsNil)
+       defer f.Close()
+       npy, err := gonpy.NewReader(f)
+       c.Assert(err, check.IsNil)
+       c.Check(npy.Shape, check.DeepEquals, []int{2, 4})
+       variants, err := npy.GetInt16()
+       if c.Check(variants, check.HasLen, 8) {
+               c.Check(variants[4:], check.DeepEquals, []int16{-1, -1, 1, 1}) // TODO: check first row too, when stable
+       }
+
+       annotations, err := ioutil.ReadFile(npydir + "/matrix.0000.annotations.csv")
+       c.Assert(err, check.IsNil)
+       c.Logf("%s", annotations)
+       for _, s := range []string{
+               "chr1:g.161A>T",
+               "chr1:g.178A>T",
+               "chr1:g.1_3delinsGGC",
+               "chr1:g.222_224del",
+       } {
+               c.Check(string(annotations), check.Matches, "(?ms).*"+s+".*")
+       }
 }