X-Git-Url: http://woldlab.caltech.edu/gitweb/?p=helpful_scripts.git;a=blobdiff_plain;f=test_translate_tsv_genes.py;fp=test_translate_tsv_genes.py;h=e4eb351683d03c09979e717153de2537554ff776;hp=0000000000000000000000000000000000000000;hb=ee247d7baddc8df0bbe82c7024631b64afdc060e;hpb=96f745635435dd777e690768ce4d02023f42701c

diff --git a/test_translate_tsv_genes.py b/test_translate_tsv_genes.py
new file mode 100644
index 0000000..e4eb351
--- /dev/null
+++ b/test_translate_tsv_genes.py
@@ -0,0 +1,15 @@
+#!/usr/bin/python3
+
+from unittest import TestCase
+from translate_tsv_genes import load_matrixes
+
+
+class TestTranslateTSV(TestCase):
+    def test_load_matrix(self):
+        """Make sure we create a matrix indexed by the union of ids
+        """
+        m = load_matrixes(['test/m1.tsv', 'test/m2.tsv'], 'FPKM')
+        self.assertEqual(m.shape, (9, 2))
+        self.assertEqual(list(m.index),
+                         ['a', 'b', 'c', 'd', 'e', 'f', 'x', 'y', 'z'])
+        self.assertEqual(list(m.keys()), ['m1.tsv', 'm2.tsv'])