def test_select_on(self):
"""
Test .select_on decorator
"""
sel = LabeledSelector(gt10)
sel.select_on('gen1')(gen1)
self.assertEqual(1, len(sel.gens))
sel_gen = sel['gen1']
#self.assertEqual(gen1.__doc__, sel_gen.__doc__)
self.assertEqual(gen1.__name__, sel_gen.__name__)
def test_stop(self):
"""
Test .stop()
"""
sel = LabeledSelector(gt10, is_even, self.labels_dict)
sel.stop()
self.assertEqual([], sel.gens)
self.assertEqual([], sel.labels)
self.assertIsNone(sel.started)
self.assertIsNone(sel.curr)
def test_select_on(self):
"""
Test .select_on decorator
"""
sel = LabeledSelector(gt10)
sel.select_on('gen1')(gen1)
self.assertEqual(1, len(sel.gens))
sel_gen = sel['gen1']
#self.assertEqual(gen1.__doc__, sel_gen.__doc__)
self.assertEqual(gen1.__name__, sel_gen.__name__)
def test_stop(self):
"""
Test .stop()
"""
sel = LabeledSelector(gt10, is_even, self.labels_dict)
sel.stop()
self.assertEqual([], sel.gens)
self.assertEqual([], sel.labels)
self.assertIsNone(sel.started)
self.assertIsNone(sel.curr)
def test_gens_dict(self):
"""
Test .gens_dict property
"""
sel = LabeledSelector(gt10, None, self.labels_dict)
expected = {val: key for key, val in self.labels_dict.iteritems()}
self.assertEqual(expected, sel.gens_dict)
def test_no_stop(self):
"""
Test LabeledSelector without pause or stop conditions
"""
sel = LabeledSelector(LabeledSelector.false, None, self.labels_dict)
expected = [{'gen1': val} for val in xrange(100)]
expected += [{'gen2': val} for val in xrange(5, 20)]
actual = self.sort_results(sel)
self.assertEqual(expected, actual)
def test_no_pause(self):
"""
Test LabeledSelector with stop_condition=gt10 and no pause condition
"""
sel = LabeledSelector(gt10, None, self.labels_dict)
expected = [{'gen1': val} for val in xrange(11)]
expected += [{'gen2': val} for val in xrange(5, 11)]
actual = self.sort_results(sel)
self.assertEqual(expected, actual)
def test_pause(self):
"""
Test LabeledSelector with stop_condition=gt10 and
pause_condition=is_even
"""
sel = LabeledSelector(gt10, is_even, self.labels_dict)
expected = [{
'gen2': 5
}, {
'gen1': 1
}, {
'gen2': 7
}, {
'gen1': 3
}, {
'gen2': 9
}, {
'gen1': 5
}, {
'gen1': 7
}, {
'gen1': 9
}]
self.assertEqual(expected, list(sel))
def test_labels_dict(self):
"""
Test .labels_dict property
"""
sel = LabeledSelector(gt10, None, self.labels_dict)
self.assertEqual(self.labels_dict, sel.labels_dict)
