initial commit

test_playback.py

@@ -0,0 +1,38 @@
+import unittest
+
+import playback
+
+
+class FramesWithLabel(unittest.TestCase):
+    def test_returns_sorted_indices_matching_label(self):
+        assignments = {5: 'walk', 1: 'walk', 3: 'idle', 8: 'walk'}
+        self.assertEqual(playback.frames_with_label(assignments, 'walk'), [1, 5, 8])
+
+    def test_ignores_other_labels(self):
+        assignments = {0: 'idle', 1: 'walk', 2: 'run'}
+        self.assertEqual(playback.frames_with_label(assignments, 'idle'), [0])
+
+    def test_no_matches_returns_empty(self):
+        assignments = {0: 'idle', 1: 'walk'}
+        self.assertEqual(playback.frames_with_label(assignments, 'jump'), [])
+
+
+class NextInCycle(unittest.TestCase):
+    def test_steps_forward(self):
+        self.assertEqual(playback.next_in_cycle([1, 5, 8], 1, 1), 5)
+
+    def test_wraps_forward_at_end(self):
+        self.assertEqual(playback.next_in_cycle([1, 5, 8], 8, 1), 1)
+
+    def test_steps_backward(self):
+        self.assertEqual(playback.next_in_cycle([1, 5, 8], 5, -1), 1)
+
+    def test_wraps_backward_at_start(self):
+        self.assertEqual(playback.next_in_cycle([1, 5, 8], 1, -1), 8)
+
+    def test_current_not_in_set_returns_first(self):
+        self.assertEqual(playback.next_in_cycle([1, 5, 8], 3, 1), 1)
+
+
+if __name__ == '__main__':
+    unittest.main()