def test_adding_batch_frame_with_outcomes_returns_new_batch_frame(self): batch_1 = Batch(frames=create_dataframe()) batch_2 = Batch(frames=create_dataframe()) batch_3 = Batch(frames=create_dataframe_same(2)) self.assertEqual(batch_3, batch_1 + batch_2)
def test_fetching_frames_by_index_should_also_return_temp_outcomes(self): batch = Batch(frames=create_dataframe_same(2), outcomes={'test': [[1], [2]]}, temp_outcomes={'test2': [[3], [4]]}) expected = Batch(frames=create_dataframe(), outcomes={'test': [[1]]}, temp_outcomes={'test2': [[3]]}) self.assertEqual(expected, batch[[0]])
def test_adding_batch_frame_with_outcomes_returns_new_batch_frame(self): batch_1 = Batch(frames=create_dataframe(), outcomes={'1': [1]}, temp_outcomes={'2': [1]}) batch_2 = Batch(frames=create_dataframe(), outcomes={'1': [2]}, temp_outcomes={'2': [2]}) batch_3 = Batch(frames=create_dataframe_same(2), outcomes={'1': [1, 2]}, temp_outcomes={'2': [1, 2]}) self.assertEqual(batch_3, batch_1 + batch_2)
def test_frames_as_numpy_array_should_frames_as_numpy_array(self): batch = Batch(frames=create_dataframe_same(2)) expected = list(np.ones((2, 1, 1))) actual = list(batch.column_as_numpy_array()) self.assertEqual(expected, actual)
def test_add_should_get_new_batch_frame_with_addition_no_outcomes(self): batch_1 = Batch(frames=create_dataframe()) batch_2 = Batch(frames=create_dataframe()) batch_3 = Batch(frames=create_dataframe_same(2)) self.assertEqual(batch_3, batch_1 + batch_2)
def test_fetching_frames_by_index(self): batch = Batch(frames=create_dataframe_same(2)) expected = Batch(frames=create_dataframe()) self.assertEqual(expected, batch[[0]])
def test_concat_batch(self): batch_1 = Batch(frames=create_dataframe()) batch_2 = Batch(frames=create_dataframe()) batch_3 = Batch(frames=create_dataframe_same(2)) self.assertEqual(batch_3, Batch.concat([batch_1, batch_2], copy=False))
def test_fetching_frames_by_index_should_raise(self): batch = Batch(frames=create_dataframe_same(2)) with self.assertRaises(TypeError): batch[1.0]
def test_fetching_frames_by_index_should_also_return_outcomes(self): batch = FrameBatch(frames=create_dataframe_same(2), outcomes={'test': [[None], [None]]}) expected = FrameBatch(frames=create_dataframe(), outcomes={'test': [[None]]}) self.assertEqual(expected, batch[[0]])