pset7 q3 code

Author: rohanrdy

week7/working/test_working.py

@@ -0,0 +1,15 @@
+import pytest
+from working import convert
+
+def test_validate_correct():
+    assert convert("9 AM to 5 PM") == "09:00 to 17:00"
+    assert convert("9:00 AM to 5:30 PM") == "09:00 to 17:30"
+    assert convert("9:00 PM to 5:30 AM") == "21:00 to 05:30"
+
+def test_value_error_to():
+    with pytest.raises(ValueError):
+        assert convert("9:00 AM lol 5:30 PM")
+
+def test_value_error_range():
+    with pytest.raises(ValueError):
+        assert convert("13:00 AM to 69:30 PM")

week7/working/working.py

@@ -0,0 +1,34 @@
+import re
+
+
+def main():
+    print(convert(input("Hours: ")))
+
+
+def convert(s):
+    if x := re.search(r"^(1?\d)(:[0-5]\d)? (AM|PM) to (1?\d)(:[0-5]\d)? (AM|PM)$", s):
+        if x.group(3) == "AM" and x.group(6) == "PM":
+            if 0 < int(x.group(1)) < 13 and 0 < int(x.group(4)) < 13:
+                p = x.group(2)
+                q = x.group(5)
+                if not p: p = ":00"
+                if not q: q = ":00"
+                if (x.group(1) and x.group(4)) == "12":
+                    return f"00{p} to {x.group(4)}{q}"
+                return f"{x.group(1).zfill(2)}{p} to {int(x.group(4))+12}{q}"
+            pass
+        elif x.group(3) == "PM" and x.group(6) == "AM":
+            if 0 < int(x.group(1)) < 13 and 0 < int(x.group(4)) < 13:
+                p = x.group(2)
+                q = x.group(5)
+                if not p: p = ":00"
+                if not q: q = ":00"
+                return f"{int(x.group(1))+12}{p} to {x.group(4).zfill(2)}{q}"
+            pass
+        raise ValueError("lol")
+    else:
+        raise ValueError("lol")
+
+
+if __name__ == "__main__":
+    main()