def test_group_single_timestep_data_into_sequences_empty_data(self): sequence = [] sequence_length = 2 with self.assertRaises(IndexError): data_utils.group_single_timestep_data_into_sequences(sequence, sequence_length)
def test_group_single_timestep_data_into_sequences_multi_item_timestep_larger_sequence(self): sequence = [[1,2],[2,3],[3,4],[4,5]] sequence_length = 4 actual = data_utils.group_single_timestep_data_into_sequences(sequence, sequence_length) expected = np.array([[1,2,2,3,3,4,4,5]]) self.assertTrue(np.array_equal(actual, expected))
def test_group_single_timestep_data_into_sequences_single_item_timestep(self): sequence = [[1],[2],[3],[4]] sequence_length = 2 actual = data_utils.group_single_timestep_data_into_sequences(sequence, sequence_length) expected = np.array([[1,2], [3,4]]) self.assertTrue(np.array_equal(actual, expected))
def test_group_single_timestep_data_into_sequences_multi_item_timestep_larger_sequence(self): sequence = [[1, 2], [2, 3], [3, 4], [4, 5]] sequence_length = 4 actual = data_utils.group_single_timestep_data_into_sequences(sequence, sequence_length) expected = np.array([[1, 2, 2, 3, 3, 4, 4, 5]]) self.assertTrue(np.array_equal(actual, expected))
def test_group_single_timestep_data_into_sequences_single_item_timestep(self): sequence = [[1], [2], [3], [4]] sequence_length = 2 actual = data_utils.group_single_timestep_data_into_sequences(sequence, sequence_length) expected = np.array([[1, 2], [3, 4]]) self.assertTrue(np.array_equal(actual, expected))
def test_group_single_timestep_data_into_sequences_empty_data(self): sequence = [] sequence_length = 2 with self.assertRaises(IndexError): data_utils.group_single_timestep_data_into_sequences(sequence, sequence_length)