def test_get_outcome_from_non_existing_udf_name_returns_empty_list(self): batch = FrameBatch(frames=create_dataframe()) self.assertEqual([], batch.get_outcomes_for('test'))
def test_get_outcome_from_non_existing_udf_name_returns_empty_list(self): batch = FrameBatch(frames=[Frame(1, np.ones((1, 1)), None)], info=None) self.assertEqual([], batch.get_outcomes_for('test'))
def test_set_outcomes_method_should_set_the_predictions_with_udf_name( self): batch = FrameBatch(frames=create_dataframe()) batch.set_outcomes('test', [None]) self.assertEqual([None], batch.get_outcomes_for('test'))
def test_set_outcomes_method_should_set_the_predictions_with_udf_name( self): batch = FrameBatch(frames=[Frame(1, np.ones((1, 1)), None)], info=None) batch.set_outcomes('test', [None]) self.assertEqual([None], batch.get_outcomes_for('test'))