Add busiest period challenge

online_judges/busiest_period/test_find_busiest_period.py

@@ -0,0 +1,28 @@
+from nose.tools import assert_equal, assert_raises
+
+
+class TestSolution(object):
+
+    def test_find_busiest_period(self):
+        solution = Solution()
+        assert_raises(TypeError, solution.find_busiest_period, None)
+        assert_equal(solution.find_busiest_period([]), None)
+        data = [
+            Data(3, 2, EventType.EXIT),
+            Data(1, 2, EventType.ENTER),
+            Data(3, 1, EventType.ENTER),
+            Data(7, 3, EventType.ENTER),
+            Data(9, 2, EventType.EXIT),
+            Data(8, 2, EventType.EXIT),
+        ]
+        assert_equal(solution.find_busiest_period(data), Period(7, 8))
+        print('Success: test_find_busiest_period')
+
+
+def main():
+    test = TestSolution()
+    test.test_find_busiest_period()
+
+
+if __name__ == '__main__':
+    main()