def test_has_outcomes_returns_true_if_the_given_name_is_in_outcomes(self):
batch = FrameBatch(frames=create_dataframe())
batch.set_outcomes('test_temp', [1], is_temp=True)
batch.set_outcomes('test', [1])
self.assertTrue(batch.has_outcome('test'))
self.assertTrue(batch.has_outcome('test_temp'))
def test_has_outcomes_returns_true_if_the_given_name_is_in_outcomes(self):
batch = FrameBatch(frames=[Frame(1, np.ones((1, 1)), None)], info=None)
batch.set_outcomes('test_temp', [1], is_temp=True)
batch.set_outcomes('test', [1])
self.assertTrue(batch.has_outcome('test'))
self.assertTrue(batch.has_outcome('test_temp'))
def test_has_outcomes_returns_false_if_the_given_name_not_in_outcomes(
self):
batch = FrameBatch(frames=create_dataframe())
self.assertFalse(batch.has_outcome('temp'))
def test_has_outcomes_returns_false_if_the_given_name_not_in_outcomes(
self):
batch = FrameBatch(frames=[Frame(1, np.ones((1, 1)), None)], info=None)
self.assertFalse(batch.has_outcome('temp'))