Datasets
Models
Diff of
/tests/limma/test_dups.py
[000000]
..
[ea0fd6]
Switch to unified view
| a | b/tests/limma/test_dups.py | ||
|---|---|---|---|
| 1 | import unittest |
||
| 2 | |||
| 3 | import numpy as np |
||
| 4 | |||
| 5 | from inmoose.limma import uniquegenelist, unwrapdups |
||
| 6 | |||
| 7 | |||
| 8 | class Test(unittest.TestCase): |
||
| 9 | def test_unwrapdups(self): |
||
| 10 | M = np.arange(12).reshape((4, 3)) |
||
| 11 | self.assertTrue(np.array_equal(unwrapdups(M, ndups=1), M)) |
||
| 12 | self.assertTrue(np.array_equal(unwrapdups(M, ndups=1, spacing=2), M)) |
||
| 13 | self.assertTrue( |
||
| 14 | np.array_equal(unwrapdups(M), [[0, 3, 1, 4, 2, 5], [6, 9, 7, 10, 8, 11]]) |
||
| 15 | ) |
||
| 16 | self.assertTrue( |
||
| 17 | np.array_equal( |
||
| 18 | unwrapdups(M, ndups=4), [[0, 3, 6, 9, 1, 4, 7, 10, 2, 5, 8, 11]] |
||
| 19 | ) |
||
| 20 | ) |
||
| 21 | self.assertTrue( |
||
| 22 | np.array_equal( |
||
| 23 | unwrapdups(M, ndups=2, spacing=2), |
||
| 24 | [[0, 6, 1, 7, 2, 8], [3, 9, 4, 10, 5, 11]], |
||
| 25 | ) |
||
| 26 | ) |
||
| 27 | |||
| 28 | def test_uniquegenelist(self): |
||
| 29 | M = np.arange(12).reshape((4, 3)) |
||
| 30 | self.assertTrue(np.array_equal(uniquegenelist(M, ndups=1), M)) |
||
| 31 | self.assertTrue(np.array_equal(uniquegenelist(M, ndups=1, spacing=2), M)) |
||
| 32 | self.assertTrue(np.array_equal(uniquegenelist(M), [[0, 1, 2], [6, 7, 8]])) |
||
| 33 | self.assertTrue(np.array_equal(uniquegenelist(M, ndups=4), [[0, 1, 2]])) |
||
| 34 | self.assertTrue( |
||
| 35 | np.array_equal( |
||
| 36 | uniquegenelist(M, ndups=2, spacing=2), [[0, 1, 2], [3, 4, 5]] |
||
| 37 | ) |
||
| 38 | ) |
||
| 39 | |||
| 40 | M = np.arange(6) |
||
| 41 | self.assertTrue(np.array_equal(uniquegenelist(M, ndups=1), M)) |
||
| 42 | self.assertTrue(np.array_equal(uniquegenelist(M, ndups=1, spacing=2), M)) |
||
| 43 | self.assertTrue(np.array_equal(uniquegenelist(M), [0, 2, 4])) |
||
| 44 | self.assertTrue(np.array_equal(uniquegenelist(M, ndups=3), [0, 3])) |
||
| 45 | self.assertTrue(np.array_equal(uniquegenelist(M, ndups=3, spacing=2), [0, 1])) |
||
| 46 | self.assertTrue( |
||
| 47 | np.array_equal(uniquegenelist(M, ndups=2, spacing=3), [0, 1, 2]) |
||
| 48 | ) |