updated unit tests

bigdatagenomics · May 29, 2019 · 2b6f2cc · 2b6f2cc
1 parent 0df2929
commit 2b6f2cc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/mango-viz/bdgenomics/mango/pileup/test/test_application.py b/mango-viz/bdgenomics/mango/pileup/test/test_application.py
@@ -22,6 +22,13 @@
 
 class MangoVizTest(unittest.TestCase):
 
+    def test_genotypes(self):
+        track = pileup.Track(viz="genotypes", label="myGenotypes", source=pileup.sources.VcfDataSource('{}'))
+
+        x = pileup.PileupViewer(locus="chr22:21340030-21340150", reference="hg19", tracks=[track])
+        assert(x.reference == 'hg19')
+        assert(x.tracks[0] == track)
+
     def test_features(self):
         track = pileup.Track(viz="features", label="myFeatures", source=pileup.sources.GA4GHFeatureJson('{}'))
 

diff --git a/mango-viz/examples/pileup-tutorial.ipynb b/mango-viz/examples/pileup-tutorial.ipynb
@@ -203,7 +203,7 @@
    ],
    "source": [
     "# make genotype track\n",
-    "tracks=[Track(viz=\"genotypes\", label=\"my genotypes\", source=pileup.sources.VcfDataSource(\"./data/genodata.v3.vcf\"))]\n",
+    "tracks=[Track(viz=\"genotypes\", label=\"my Genotypes\", source=pileup.sources.VcfDataSource(\"./data/genodata.v3.vcf\"))]\n",
     "\n",
     "# render tracks in widget\n",
     "genotypes = pileup.PileupViewer(locus=\"chr22:21340030-21340150\", reference=\"hg19\", tracks=tracks)\n",