Esempio n. 1
0
 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'))
Esempio n. 2
0
 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'))
Esempio n. 3
0
 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'))
Esempio n. 4
0
 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'))