-
-
Notifications
You must be signed in to change notification settings - Fork 216
/
test_arraytable.py
42 lines (29 loc) · 1.54 KB
/
test_arraytable.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import unittest
import numpy as np
from ete4 import ArrayTable
from .. import datasets as ds
class Test_Core_ArrayTable(unittest.TestCase):
"""Tests reading clustering or phylogenetic profile data."""
def test_arraytable_parser(self):
"""Test reading numeric tables."""
A = ArrayTable(ds.expression)
self.assertEqual(A.get_row_vector("A").tolist(),
[-1.23, -0.81, 1.79, 0.78,-0.42,-0.69, 0.58])
self.assertEqual(A.get_several_row_vectors(["A", "C"]).tolist(),
[[-1.23, -0.81, 1.79, 0.78, -0.42, -0.69, 0.58],
[-2.19, 0.13, 0.65, -0.51, 0.52, 1.04, 0.36]])
self.assertEqual(A.get_several_column_vectors(["col2", "col7"]).tolist(),
[[-0.81, -0.94, 0.13, -0.98, -0.83, -1.11, -1.17, -1.25],
[0.58, 1.12, 0.36, 0.93, 0.65, 0.48, 0.26, 0.77]])
self.assertEqual(A.get_column_vector("col4").tolist(),
[0.78, 0.36, -0.51, -0.76, 0.07, -0.14, 0.23, -0.3])
A.remove_column("col4")
self.assertTrue(A.get_column_vector("col4") is None)
Abis = A.merge_columns({"merged1": ["col1", "col2"],
"merged2": ["col5", "col6"]}, "mean")
self.assertTrue((np.round(Abis.get_column_vector("merged1"), 3) ==
[-1.02, -1.35, -1.03, -1.1, -1.15, -1.075, -1.37, -1.39]).all())
# Continue this......
# Sure. C'mon Jaime, you can do it!
if __name__ == '__main__':
unittest.main()