def testEncode(self): events = [0, 1, 0, 2, 0] sequence_example = self.enc.encode(events) expected_inputs = [[0], [1], [0], [2]] expected_labels = [1, 0, 2, 0] expected_sequence_example = encoder_decoder.make_sequence_example( expected_inputs, expected_labels) self.assertEqual(sequence_example, expected_sequence_example)
def testEncode(self): control_events = [1, 1, 1, 0, 0] target_events = [0, 1, 0, 2, 0] sequence_example = self.enc.encode(control_events, target_events) expected_inputs = [[0.0, 1.0, 1.0, 0.0, 0.0], [0.0, 1.0, 0.0, 1.0, 0.0], [1.0, 0.0, 1.0, 0.0, 0.0], [1.0, 0.0, 0.0, 0.0, 1.0]] expected_labels = [1, 0, 2, 0] expected_sequence_example = encoder_decoder.make_sequence_example( expected_inputs, expected_labels) self.assertEqual(sequence_example, expected_sequence_example)
def testEncode(self): events = [100, 100, 107, 111, NO_EVENT, 99, 112, NOTE_OFF, NO_EVENT] melody = melodies_lib.Melody(events) melody.squash(self.min_note, self.max_note, self.transpose_to_key) sequence_example = self.med.encode(melody) expected_inputs = [[ 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0 ], [ 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0 ], [ 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0 ], [ 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0 ], [ 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0 ], [ 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0 ], [ 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0 ], [ 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0 ]] expected_labels = [2, 9, 13, 0, 13, 2, 1, 0] expected_sequence_example = encoder_decoder.make_sequence_example( expected_inputs, expected_labels) self.assertEqual(sequence_example, expected_sequence_example)