+#!/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'])