Add time api and test

2022-10-22 14:53:18 +08:00
parent e3d3021286
commit 15ed42e5ac
6 changed files with 187 additions and 2 deletions
--- a/test_tjnetwork.py
+++ b/test_tjnetwork.py
@@ -1500,6 +1500,102 @@ class TestApi:
        self.leave(p)


+    def test_time(self):
+        p = 'test_time'
+        self.enter(p)
+
+        t = get_time(p)
+        assert t['DURATION'] == '0:00'
+        assert t['HYDRAULIC TIMESTEP'] == '1:00'
+        assert t['QUALITY TIMESTEP'] == '0:05'
+        assert t['RULE TIMESTEP'] == '0:05'
+        assert t['PATTERN TIMESTEP'] == '1:00'
+        assert t['PATTERN START'] == '0:00'
+        assert t['REPORT TIMESTEP'] == '1:00'
+        assert t['REPORT START'] == '0:00'
+        assert t['START CLOCKTIME'] == '12:00 AM'
+        assert t['STATISTIC'] == TIME_STATISTIC_NONE
+
+        t['STATISTIC'] = TIME_STATISTIC_AVERAGED
+        set_time(p, ChangeSet(t))
+
+        t = get_time(p)
+        assert t['DURATION'] == '0:00'
+        assert t['HYDRAULIC TIMESTEP'] == '1:00'
+        assert t['QUALITY TIMESTEP'] == '0:05'
+        assert t['RULE TIMESTEP'] == '0:05'
+        assert t['PATTERN TIMESTEP'] == '1:00'
+        assert t['PATTERN START'] == '0:00'
+        assert t['REPORT TIMESTEP'] == '1:00'
+        assert t['REPORT START'] == '0:00'
+        assert t['START CLOCKTIME'] == '12:00 AM'
+        assert t['STATISTIC'] == TIME_STATISTIC_AVERAGED
+
+        self.leave(p)
+
+
+    def test_time_op(self):
+        p = 'test_time_op'
+        self.enter(p)
+
+        t = get_time(p)
+        assert t['DURATION'] == '0:00'
+        assert t['HYDRAULIC TIMESTEP'] == '1:00'
+        assert t['QUALITY TIMESTEP'] == '0:05'
+        assert t['RULE TIMESTEP'] == '0:05'
+        assert t['PATTERN TIMESTEP'] == '1:00'
+        assert t['PATTERN START'] == '0:00'
+        assert t['REPORT TIMESTEP'] == '1:00'
+        assert t['REPORT START'] == '0:00'
+        assert t['START CLOCKTIME'] == '12:00 AM'
+        assert t['STATISTIC'] == TIME_STATISTIC_NONE
+
+        t['STATISTIC'] = TIME_STATISTIC_AVERAGED
+        cs = set_time(p, ChangeSet(t)).operations[0]
+        assert cs['operation'] == API_UPDATE
+        assert cs['type'] == 'time'
+        assert cs['DURATION'] == '0:00'
+        assert cs['HYDRAULIC TIMESTEP'] == '1:00'
+        assert cs['QUALITY TIMESTEP'] == '0:05'
+        assert cs['RULE TIMESTEP'] == '0:05'
+        assert cs['PATTERN TIMESTEP'] == '1:00'
+        assert cs['PATTERN START'] == '0:00'
+        assert cs['REPORT TIMESTEP'] == '1:00'
+        assert cs['REPORT START'] == '0:00'
+        assert cs['START CLOCKTIME'] == '12:00 AM'
+        assert cs['STATISTIC'] == TIME_STATISTIC_AVERAGED
+
+        cs = execute_undo(p).operations[0]
+        assert cs['operation'] == API_UPDATE
+        assert cs['type'] == 'time'
+        assert cs['DURATION'] == '0:00'
+        assert cs['HYDRAULIC TIMESTEP'] == '1:00'
+        assert cs['QUALITY TIMESTEP'] == '0:05'
+        assert cs['RULE TIMESTEP'] == '0:05'
+        assert cs['PATTERN TIMESTEP'] == '1:00'
+        assert cs['PATTERN START'] == '0:00'
+        assert cs['REPORT TIMESTEP'] == '1:00'
+        assert cs['REPORT START'] == '0:00'
+        assert cs['START CLOCKTIME'] == '12:00 AM'
+        assert cs['STATISTIC'] == TIME_STATISTIC_NONE
+
+        cs = execute_redo(p).operations[0]
+        assert cs['operation'] == API_UPDATE
+        assert cs['type'] == 'time'
+        assert cs['DURATION'] == '0:00'
+        assert cs['HYDRAULIC TIMESTEP'] == '1:00'
+        assert cs['QUALITY TIMESTEP'] == '0:05'
+        assert cs['RULE TIMESTEP'] == '0:05'
+        assert cs['PATTERN TIMESTEP'] == '1:00'
+        assert cs['PATTERN START'] == '0:00'
+        assert cs['REPORT TIMESTEP'] == '1:00'
+        assert cs['REPORT START'] == '0:00'
+        assert cs['START CLOCKTIME'] == '12:00 AM'
+        assert cs['STATISTIC'] == TIME_STATISTIC_AVERAGED
+
+        self.leave(p)
+
+
    def test_snapshot(self):
        p = "test_snapshot"
        self.enter(p)