Add emitter api and test

test_tjnetwork.py

@@ -1187,6 +1187,7 @@ class TestApi:
 
         set_demand(p, ChangeSet({'junction': 'j1', 'demands': [{'demand': 10.0, 'pattern': None, 'category': 'x'},
                                                                {'demand': 20.0, 'pattern': None, 'category': None}]}))
+
         d = get_demand(p, 'j1')
         assert d['junction'] == 'j1'
         ds = d['demands']
@@ -1199,6 +1200,7 @@ class TestApi:
         assert ds[1]['category'] == None
 
         set_demand(p, ChangeSet({'junction': 'j1', 'demands': []}))
+
         d = get_demand(p, 'j1')
         assert d['junction'] == 'j1'
         assert d['demands'] == []
@@ -1444,6 +1446,60 @@ class TestApi:
         self.leave(p)
 
 
+    def test_emitter(self):
+        p = 'test_emitter'
+        self.enter(p)
+
+        add_junction(p, ChangeSet({'id': 'j1', 'x': 0.0, 'y': 10.0, 'elevation': 20.0}))
+        assert is_junction(p, 'j1')
+
+        e = get_emitter(p, 'j1')
+        assert e['junction'] == 'j1'
+        assert e['coefficient'] == None
+
+        set_emitter(p, ChangeSet({'junction': 'j1', 'coefficient': 10.0}))
+
+        e = get_emitter(p, 'j1')
+        assert e['junction'] == 'j1'
+        assert e['coefficient'] == 10.0
+
+        set_emitter(p, ChangeSet({'junction': 'j1', 'coefficient': None}))
+
+        e = get_emitter(p, 'j1')
+        assert e['junction'] == 'j1'
+        assert e['coefficient'] == None
+
+        self.leave(p)
+
+
+    def test_emitter_op(self):
+        p = 'test_emitter_op'
+        self.enter(p)
+
+        add_junction(p, ChangeSet({'id': 'j1', 'x': 0.0, 'y': 10.0, 'elevation': 20.0}))
+        assert is_junction(p, 'j1')
+
+        cs = set_emitter(p, ChangeSet({'junction': 'j1', 'coefficient': 10.0})).operations[0]
+        assert cs['operation'] == API_UPDATE
+        assert cs['type'] == 'emitter'
+        assert cs['junction'] == 'j1'
+        assert cs['coefficient'] == 10.0
+
+        cs = execute_undo(p).operations[0]
+        assert cs['operation'] == API_UPDATE
+        assert cs['type'] == 'emitter'
+        assert cs['junction'] == 'j1'
+        assert cs['coefficient'] == None
+
+        cs = execute_redo(p).operations[0]
+        assert cs['operation'] == API_UPDATE
+        assert cs['type'] == 'emitter'
+        assert cs['junction'] == 'j1'
+        assert cs['coefficient'] == 10.0
+
+        self.leave(p)
+
+
     def test_snapshot(self):
         p = "test_snapshot"
         self.enter(p)