class TestStaff(unittest.TestCase): def setUp(self): self.staff = Staff() def testEmptyStaff(self): self.assertEqual(self.staff.numMeasures(), 0) self.assertEqual(len(self.staff), 0) def testAddMeasure(self): self.staff.addMeasure(Measure(1)) self.assertEqual(self.staff.numMeasures(), 1) self.assertEqual(len(self.staff), 1) self.staff.addMeasure(Measure(2)) self.assertEqual(self.staff.numMeasures(), 2) self.assertEqual(len(self.staff), 3) def testIterateOverMeasures(self): for i in range(1, 16): self.staff.addMeasure(Measure(i)) for i, measure in enumerate(self.staff): self.assertEqual(len(measure), i + 1) def testGetMeasureByIndex(self): for i in range(1, 16): self.staff.addMeasure(Measure(i)) for i in range(1, 16): self.assertEqual(len(self.staff[i - 1]), i) def testDeleteLastMeasure(self): self.staff.addMeasure(Measure(1)) self.staff.addMeasure(Measure(2)) self.staff.addMeasure(Measure(3)) self.staff.addMeasure(Measure(4)) self.assertEqual(self.staff.numMeasures(), 4) self.staff.deleteLastMeasure() self.assertEqual(self.staff.numMeasures(), 3) for index in range(1, 4): self.assertEqual(len(self.staff[index - 1]), index) def testDeleteMeasure(self): self.staff.addMeasure(Measure(1)) self.staff.addMeasure(Measure(2)) self.staff.deleteMeasure(NotePosition(measureIndex = 0)) self.assertEqual(self.staff.numMeasures(), 1) self.assertEqual(len(self.staff), 2) def testDeleteMeasure_BadIndex(self): self.staff.addMeasure(Measure(1)) self.staff.addMeasure(Measure(2)) self.assertRaises(BadTimeError, self.staff.deleteMeasure, NotePosition(measureIndex = -1)) self.assertRaises(BadTimeError, self.staff.deleteMeasure, NotePosition(measureIndex = 2)) def testInsertMeasure(self): self.staff.addMeasure(Measure(1)) self.staff.addMeasure(Measure(3)) self.staff.insertMeasure(NotePosition(measureIndex = 1), Measure(2)) for i, measure in enumerate(self.staff): self.assertEqual(len(measure), i + 1) def testInsertMeasureAtStart(self): self.staff.addMeasure(Measure(2)) self.staff.insertMeasure(NotePosition(measureIndex = 0), Measure(1)) for i, measure in enumerate(self.staff): self.assertEqual(len(measure), i + 1) def testInsertMeasureAtEnd(self): self.staff.addMeasure(Measure(1)) self.staff.insertMeasure(NotePosition(measureIndex = 1), Measure(2)) for i, measure in enumerate(self.staff): self.assertEqual(len(measure), i + 1) def testInsertMeasure_BadIndex(self): self.staff.addMeasure(Measure(1)) self.staff.addMeasure(Measure(2)) m = Measure(3) self.assertRaises(BadTimeError, self.staff.insertMeasure, NotePosition(measureIndex = -1), m) self.assertRaises(BadTimeError, self.staff.insertMeasure, NotePosition(measureIndex = 3), m) def testClearStaff(self): self.staff.addMeasure(Measure(16)) self.staff.addMeasure(Measure(16)) self.staff.addMeasure(Measure(16)) self.staff.addMeasure(Measure(16)) self.staff.clear() self.assertEqual(len(self.staff), 0) self.assertEqual(self.staff.numMeasures(), 0)
class TestStaff(unittest.TestCase): def setUp(self): self.staff = Staff() def testEmptyStaff(self): self.assertEqual(self.staff.numMeasures(), 0) self.assertEqual(len(self.staff), 0) def testAddMeasure(self): self.staff.addMeasure(Measure(1)) self.assertEqual(self.staff.numMeasures(), 1) self.assertEqual(len(self.staff), 1) self.staff.addMeasure(Measure(2)) self.assertEqual(self.staff.numMeasures(), 2) self.assertEqual(len(self.staff), 3) def testIterateOverMeasures(self): for i in range(1, 16): self.staff.addMeasure(Measure(i)) for i, measure in enumerate(self.staff): self.assertEqual(len(measure), i + 1) def testGetMeasureByIndex(self): for i in range(1, 16): self.staff.addMeasure(Measure(i)) for i in range(1, 16): self.assertEqual(len(self.staff[i - 1]), i) def testDeleteLastMeasure(self): self.staff.addMeasure(Measure(1)) self.staff.addMeasure(Measure(2)) self.staff.addMeasure(Measure(3)) self.staff.addMeasure(Measure(4)) self.assertEqual(self.staff.numMeasures(), 4) self.staff.deleteLastMeasure() self.assertEqual(self.staff.numMeasures(), 3) for index in range(1, 4): self.assertEqual(len(self.staff[index - 1]), index) def testDeleteMeasure(self): self.staff.addMeasure(Measure(1)) self.staff.addMeasure(Measure(2)) self.staff.deleteMeasure(NotePosition(measureIndex=0)) self.assertEqual(self.staff.numMeasures(), 1) self.assertEqual(len(self.staff), 2) def testDeleteMeasure_BadIndex(self): self.staff.addMeasure(Measure(1)) self.staff.addMeasure(Measure(2)) self.assertRaises(BadTimeError, self.staff.deleteMeasure, NotePosition(measureIndex=-1)) self.assertRaises(BadTimeError, self.staff.deleteMeasure, NotePosition(measureIndex=2)) def testInsertMeasure(self): self.staff.addMeasure(Measure(1)) self.staff.addMeasure(Measure(3)) self.staff.insertMeasure(NotePosition(measureIndex=1), Measure(2)) for i, measure in enumerate(self.staff): self.assertEqual(len(measure), i + 1) def testInsertMeasureAtStart(self): self.staff.addMeasure(Measure(2)) self.staff.insertMeasure(NotePosition(measureIndex=0), Measure(1)) for i, measure in enumerate(self.staff): self.assertEqual(len(measure), i + 1) def testInsertMeasureAtEnd(self): self.staff.addMeasure(Measure(1)) self.staff.insertMeasure(NotePosition(measureIndex=1), Measure(2)) for i, measure in enumerate(self.staff): self.assertEqual(len(measure), i + 1) def testInsertMeasure_BadIndex(self): self.staff.addMeasure(Measure(1)) self.staff.addMeasure(Measure(2)) m = Measure(3) self.assertRaises(BadTimeError, self.staff.insertMeasure, NotePosition(measureIndex=-1), m) self.assertRaises(BadTimeError, self.staff.insertMeasure, NotePosition(measureIndex=3), m) def testClearStaff(self): self.staff.addMeasure(Measure(16)) self.staff.addMeasure(Measure(16)) self.staff.addMeasure(Measure(16)) self.staff.addMeasure(Measure(16)) self.staff.clear() self.assertEqual(len(self.staff), 0) self.assertEqual(self.staff.numMeasures(), 0)
class TestCallBack(unittest.TestCase): def setUp(self): self.staff = Staff() self.staff.addMeasure(Measure(16)) self.calls = [] def myCallBack(position): self.calls.append((position.measureIndex, position.noteTime, position.drumIndex)) self.staff.setCallBack(myCallBack) def testAddNoteCallBack(self): np = NotePosition(measureIndex = 0, noteTime = 0, drumIndex = 0) self.staff.addNote(np, "x") self.assertEqual(len(self.calls), 1) self.assertEqual(self.calls[0], (0, 0, 0)) self.staff.addNote(np, "x") self.assertEqual(len(self.calls), 1) self.staff.addNote(np, "o") self.assertEqual(len(self.calls), 2) self.assertEqual(self.calls[1], (0, 0, 0)) def testDeleteNoteCallBack(self): np = NotePosition(measureIndex = 0, noteTime = 0, drumIndex = 0) self.staff.deleteNote(np) self.assertEqual(len(self.calls), 0) self.staff.addNote(np, "x") self.assertEqual(len(self.calls), 1) self.assertEqual(self.calls[0], (0, 0, 0)) self.staff.deleteNote(np) self.assertEqual(len(self.calls), 2) self.assertEqual(self.calls[1], (0, 0, 0)) def testToggleNoteCallBack(self): np = NotePosition(measureIndex = 0, noteTime = 0, drumIndex = 0) self.staff.toggleNote(np, "x") self.assertEqual(len(self.calls), 1) self.assertEqual(self.calls[0], (0, 0, 0)) self.staff.toggleNote(np, "x") self.assertEqual(len(self.calls), 2) self.assertEqual(self.calls[1], (0, 0, 0)) self.staff.toggleNote(np, "x") self.assertEqual(len(self.calls), 3) self.assertEqual(self.calls[2], (0, 0, 0)) self.staff.toggleNote(np, "o") self.assertEqual(len(self.calls), 4) self.assertEqual(self.calls[2], (0, 0, 0)) def testClearCallBack(self): self.staff.clearCallBack() self.staff.addNote(NotePosition(measureIndex = 0, noteTime = 0, drumIndex = 0), "x") self.assertEqual(len(self.calls), 0) def testAddMeasureCallBack(self): np = NotePosition(measureIndex = 1, noteTime = 0, drumIndex = 0) self.staff.addMeasure(Measure(16)) self.staff.addNote(np, "x") self.assertEqual(len(self.calls), 1) self.assertEqual(self.calls[0], (1, 0, 0)) def testDeleteMeasureCallBack(self): self.staff.addMeasure(Measure(16)) self.staff.addMeasure(Measure(16)) np = NotePosition(measureIndex = 1, noteTime = 0, drumIndex = 0) self.staff.addNote(np, "x") self.staff.deleteMeasure(NotePosition(measureIndex = 0)) self.staff.deleteNote(NotePosition(measureIndex = 0, noteTime = 0, drumIndex = 0)) self.staff.addNote(np, "x") self.assertEqual(len(self.calls), 3) self.assertEqual(self.calls[0], (1, 0, 0)) self.assertEqual(self.calls[1], (0, 0, 0)) self.assertEqual(self.calls[2], (1, 0, 0)) def testInsertMeasureCallBack(self): np0 = NotePosition(measureIndex = 0, noteTime = 0, drumIndex = 0) np1 = NotePosition(measureIndex = 1, noteTime = 0, drumIndex = 0) self.staff.addNote(np0, "x") self.staff.insertMeasure(np0, Measure(8)) self.staff.addNote(np0, "x") self.staff.deleteNote(np1) self.assertEqual(len(self.calls), 3) self.assertEqual(self.calls[0], (0, 0, 0)) self.assertEqual(self.calls[1], (0, 0, 0)) self.assertEqual(self.calls[2], (1, 0, 0))
class TestCallBack(unittest.TestCase): def setUp(self): self.staff = Staff() self.staff.addMeasure(Measure(16)) self.calls = [] def myCallBack(position): self.calls.append( (position.measureIndex, position.noteTime, position.drumIndex)) self.staff.setCallBack(myCallBack) def testAddNoteCallBack(self): np = NotePosition(measureIndex=0, noteTime=0, drumIndex=0) self.staff.addNote(np, "x") self.assertEqual(len(self.calls), 1) self.assertEqual(self.calls[0], (0, 0, 0)) self.staff.addNote(np, "x") self.assertEqual(len(self.calls), 1) self.staff.addNote(np, "o") self.assertEqual(len(self.calls), 2) self.assertEqual(self.calls[1], (0, 0, 0)) def testDeleteNoteCallBack(self): np = NotePosition(measureIndex=0, noteTime=0, drumIndex=0) self.staff.deleteNote(np) self.assertEqual(len(self.calls), 0) self.staff.addNote(np, "x") self.assertEqual(len(self.calls), 1) self.assertEqual(self.calls[0], (0, 0, 0)) self.staff.deleteNote(np) self.assertEqual(len(self.calls), 2) self.assertEqual(self.calls[1], (0, 0, 0)) def testToggleNoteCallBack(self): np = NotePosition(measureIndex=0, noteTime=0, drumIndex=0) self.staff.toggleNote(np, "x") self.assertEqual(len(self.calls), 1) self.assertEqual(self.calls[0], (0, 0, 0)) self.staff.toggleNote(np, "x") self.assertEqual(len(self.calls), 2) self.assertEqual(self.calls[1], (0, 0, 0)) self.staff.toggleNote(np, "x") self.assertEqual(len(self.calls), 3) self.assertEqual(self.calls[2], (0, 0, 0)) self.staff.toggleNote(np, "o") self.assertEqual(len(self.calls), 4) self.assertEqual(self.calls[2], (0, 0, 0)) def testClearCallBack(self): self.staff.clearCallBack() self.staff.addNote( NotePosition(measureIndex=0, noteTime=0, drumIndex=0), "x") self.assertEqual(len(self.calls), 0) def testAddMeasureCallBack(self): np = NotePosition(measureIndex=1, noteTime=0, drumIndex=0) self.staff.addMeasure(Measure(16)) self.staff.addNote(np, "x") self.assertEqual(len(self.calls), 1) self.assertEqual(self.calls[0], (1, 0, 0)) def testDeleteMeasureCallBack(self): self.staff.addMeasure(Measure(16)) self.staff.addMeasure(Measure(16)) np = NotePosition(measureIndex=1, noteTime=0, drumIndex=0) self.staff.addNote(np, "x") self.staff.deleteMeasure(NotePosition(measureIndex=0)) self.staff.deleteNote( NotePosition(measureIndex=0, noteTime=0, drumIndex=0)) self.staff.addNote(np, "x") self.assertEqual(len(self.calls), 3) self.assertEqual(self.calls[0], (1, 0, 0)) self.assertEqual(self.calls[1], (0, 0, 0)) self.assertEqual(self.calls[2], (1, 0, 0)) def testInsertMeasureCallBack(self): np0 = NotePosition(measureIndex=0, noteTime=0, drumIndex=0) np1 = NotePosition(measureIndex=1, noteTime=0, drumIndex=0) self.staff.addNote(np0, "x") self.staff.insertMeasure(np0, Measure(8)) self.staff.addNote(np0, "x") self.staff.deleteNote(np1) self.assertEqual(len(self.calls), 3) self.assertEqual(self.calls[0], (0, 0, 0)) self.assertEqual(self.calls[1], (0, 0, 0)) self.assertEqual(self.calls[2], (1, 0, 0))