add test case to make sure we merge tables with different sets of labels correctly

diff --git a/test/m1.tsv b/test/m1.tsv
new file mode 100644 (file)
index 0000000..ae50984
--- /dev/null
+++ b/test/m1.tsv
@@ -0,0 +1,7 @@
+genes  FPKM
+a      1
+b      2
+c      0
+d      4
+e      0
+f      6

diff --git a/test/m2.tsv b/test/m2.tsv
new file mode 100644 (file)
index 0000000..0b14d28
--- /dev/null
+++ b/test/m2.tsv
@@ -0,0 +1,7 @@
+genes  FPKM
+a      10
+b      20
+c      0
+x      40
+y      50
+z      0

diff --git a/test_translate_tsv_genes.py b/test_translate_tsv_genes.py
new file mode 100644 (file)
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'])