Add test of times.py Answers UCL-RITS/rsd-classwork-2020#61

liamchalcroft · Nov 4, 2020 · b3abfdb · b3abfdb
1 parent 1642a86
commit b3abfdb
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/week05-testing/test_times.py b/week05-testing/test_times.py
@@ -0,0 +1,9 @@
+import times
+
+def test_given_input():
+    large = times.time_range("2010-01-12 10:00:00", "2010-01-12 12:00:00")
+    short = times.time_range("2010-01-12 10:30:00", "2010-01-12 10:45:00", 2, 60)
+    result = times.compute_overlap_time(large, short)
+    expected = [('2010-01-12 10:30:00', '2010-01-12 10:37:00'),
+                 ('2010-01-12 10:38:00', '2010-01-12 10:45:00')]
+    assert result == expected