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_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_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)