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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
#!/usr/bin/env python
# -----------------------------------------------------------------------------
# Copyright (c) 2011-2017, The BIOM Format Development Team.
#
# Distributed under the terms of the Modified BSD License.
#
# The full license is in the file COPYING.txt, distributed with this software.
# -----------------------------------------------------------------------------
from unittest import TestCase, main
import numpy as np
import biom
from biom.cli.uc_processor import _from_uc
from biom.tests.long_lines import uc_minimal, uc
class TestUcProcessor(TestCase):
def setUp(self):
"""Set up data for use in unit tests."""
self.cmd = _from_uc
self.uc_minimal = uc_minimal.split('\n')
self.uc = uc.split('\n')
self.rep_set = rep_set.split('\n')
self.rep_set_no_mapping = rep_set_no_mapping.split('\n')
self.rep_set_missing_id = rep_set_missing_id.split('\n')
def test_basic(self):
obs = self.cmd(self.uc_minimal)
expected = biom.Table(np.array([[1.0]]),
observation_ids=['f2_1539'],
sample_ids=['f2'])
self.assertEqual(obs, expected)
def test_basic_w_mapping(self):
obs = self.cmd(self.uc_minimal, self.rep_set)
expected = biom.Table(np.array([[1.0]]),
observation_ids=['otu1'],
sample_ids=['f2'])
self.assertEqual(obs, expected)
def test_rep_set_no_mapping(self):
self.assertRaises(ValueError, self.cmd, self.uc_minimal,
self.rep_set_no_mapping)
def test_rep_set_missing_id(self):
self.assertRaises(ValueError, self.cmd, self.uc_minimal,
self.rep_set_missing_id)
def test_uc(self):
obs = self.cmd(self.uc)
expected = biom.Table(np.array([[1.0, 1.0], [0.0, 1.0]]),
observation_ids=['f2_1539', 'f3_1540'],
sample_ids=['f2', 'f3'])
self.assertEqual(obs, expected)
def test_uc_w_mapping(self):
obs = self.cmd(self.uc, self.rep_set)
expected = biom.Table(np.array([[1.0, 1.0], [0.0, 1.0]]),
observation_ids=['otu1', 'otu2'],
sample_ids=['f2', 'f3'])
self.assertEqual(obs, expected)
rep_set = """>otu1 f2_1539
ACGT
>otu2 f3_1540
ACCT
"""
rep_set_no_mapping = """>otu1
ACGT
>otu2
ACCT
"""
rep_set_missing_id = """>otu1 f99_1539
ACGT
>otu2 f99_1539
ACCT
"""
if __name__ == '__main__':
main()
|
