moderated input validation based on sample solution

Author: shisei12uni

test_times.py

@@ -40,4 +40,10 @@ def test_input_ends_at_start():
 def test_input_validation():
     """input validation, does code raise error when start time > end time?"""
     with raises(ValueError):
-        time_range("2010-01-12 12:00:00", "2010-01-12 10:00:00")
+        time_range("2010-01-12 12:00:00", "2010-01-12 10:00:00")
+    """
+    sample solution is
+    with pytest.raises(ValueError) as e:
+        time_range("2010-01-12 10:00:00", "2010-01-12 09:30:00")
+    assert e.match("The end of the time range has to come strictly after its start)
+    """

times.py

@@ -3,8 +3,8 @@
 def time_range(start_time, end_time, number_of_intervals=1, gap_between_intervals_s=0):
     start_time_s = datetime.datetime.strptime(start_time, "%Y-%m-%d %H:%M:%S")
     end_time_s = datetime.datetime.strptime(end_time, "%Y-%m-%d %H:%M:%S")
-    if end_time_s < start_time_s:
-        raise ValueError ("end_time is smaller than start_time")
+    if end_time_s <= start_time_s:
+        raise ValueError ("the end_time has to come after the start_time")
     d = (end_time_s - start_time_s).total_seconds() / number_of_intervals + gap_between_intervals_s * (1 / number_of_intervals - 1)
     sec_range = [(start_time_s + datetime.timedelta(seconds=i * d + i * gap_between_intervals_s),
                   start_time_s + datetime.timedelta(seconds=(i + 1) * d + i * gap_between_intervals_s))