From ee247d7baddc8df0bbe82c7024631b64afdc060e Mon Sep 17 00:00:00 2001 From: Diane Trout Date: Mon, 4 May 2015 15:24:16 -0700 Subject: [PATCH] add test case to make sure we merge tables with different sets of labels correctly --- test/m1.tsv | 7 +++++++ test/m2.tsv | 7 +++++++ test_translate_tsv_genes.py | 15 +++++++++++++++ 3 files changed, 29 insertions(+) create mode 100644 test/m1.tsv create mode 100644 test/m2.tsv create mode 100644 test_translate_tsv_genes.py diff --git a/test/m1.tsv b/test/m1.tsv new file mode 100644 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 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 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']) -- 2.30.2