def test_getFlow(self):
"""FlowgramCollection.getFlow should return specified flow"""
a = [('a','0.5 1.0 4.0 0.0 1.5 0.0 0.0 2.0'),
('b','1.5 1.0 0.0 0.0 2.5 1.0 2.0 1.0'),
('c','2.5 0.0 4.0 0.0 0.5 1.0 0.0 1.0')]
f = FlowgramCollection(a)
self.assertEqual(f.getFlow('a'), '0.5 1.0 4.0 0.0 1.5 0.0 0.0 2.0')
self.assertRaises(KeyError, f.getFlow, 'd')
def test_getFlow(self):
"""FlowgramCollection.getFlow should return specified flow"""
a = [('a','0.5 1.0 4.0 0.0 1.5 0.0 0.0 2.0'),
('b','1.5 1.0 0.0 0.0 2.5 1.0 2.0 1.0'),
('c','2.5 0.0 4.0 0.0 0.5 1.0 0.0 1.0')]
f = FlowgramCollection(a)
self.assertEqual(f.getFlow('a'), '0.5 1.0 4.0 0.0 1.5 0.0 0.0 2.0')
self.assertRaises(KeyError, f.getFlow, 'd')
def test_getFlow(self):
"""FlowgramCollection.getFlow should return specified flow"""
a = [
("a", "0.5 1.0 4.0 0.0 1.5 0.0 0.0 2.0"),
("b", "1.5 1.0 0.0 0.0 2.5 1.0 2.0 1.0"),
("c", "2.5 0.0 4.0 0.0 0.5 1.0 0.0 1.0"),
]
f = FlowgramCollection(a)
self.assertEqual(f.getFlow("a"), "0.5 1.0 4.0 0.0 1.5 0.0 0.0 2.0")
self.assertRaises(KeyError, f.getFlow, "d")
def test_average_flowgrams(self):
"""_average_flowgrams computes an averaged flowgram for each cluster."""
fc = FlowgramCollection({'a':'1.0 0.0 0.0 1.0 1.0 1.2 1.2 0.8',
'b':'1.2 1.0 0.0 0.8 1.2 2.4 1.0 0.0'})
#return the centroid unmodified if sample_mapping = 1
actual = list(_average_flowgrams({'a':'b'}, fc, {'a':['a']}))
self.assertEqual(actual , [(fc.getFlow('a'), 'a')])
actual = list(_average_flowgrams({'a':'b'}, fc, {'a':['a','b']}))
self.assertEqual(actual , [(Flowgram(['1.1 0.5 0.0 0.9 1.1 1.8 1.1 0.4']), 'a')])
def test_average_flowgrams(self):
"""_average_flowgrams computes an averaged flowgram for each cluster."""
fc = FlowgramCollection({
'a': '1.0 0.0 0.0 1.0 1.0 1.2 1.2 0.8',
'b': '1.2 1.0 0.0 0.8 1.2 2.4 1.0 0.0'
})
#return the centroid unmodified if sample_mapping = 1
actual = list(_average_flowgrams({'a': 'b'}, fc, {'a': ['a']}))
self.assertEqual(actual, [(fc.getFlow('a'), 'a')])
actual = list(_average_flowgrams({'a': 'b'}, fc, {'a': ['a', 'b']}))
self.assertEqual(
actual, [(Flowgram(['1.1 0.5 0.0 0.9 1.1 1.8 1.1 0.4']), 'a')])
