Prevent infinite loop

test_tjnetwork.py

@@ -6097,24 +6097,81 @@ class TestApi:
 
         cs = generate_district_metering_area(p, 3).operations
         assert len(cs) == 3
+        assert cs[0]['operation'] == API_ADD
+        assert cs[0]['type'] == 'district_metering_area'
+        assert cs[0]['id'] == 'DMA_1_1'
+        assert cs[1]['operation'] == API_ADD
+        assert cs[1]['type'] == 'district_metering_area'
+        assert cs[1]['id'] == 'DMA_1_2'
+        assert cs[2]['operation'] == API_ADD
+        assert cs[2]['type'] == 'district_metering_area'
+        assert cs[2]['id'] == 'DMA_1_3'
 
         dmas = get_all_district_metering_area_ids(p)
         assert len(dmas) == 3
+        assert dmas[0] == 'DMA_1_1'
+        assert dmas[1] == 'DMA_1_2'
+        assert dmas[2] == 'DMA_1_3'
 
         cs = generate_district_metering_area(p, 3).operations
         assert len(cs) == 6
+        assert cs[0]['operation'] == API_DELETE
+        assert cs[0]['type'] == 'district_metering_area'
+        assert cs[0]['id'] == 'DMA_1_1'
+        assert cs[1]['operation'] == API_DELETE
+        assert cs[1]['type'] == 'district_metering_area'
+        assert cs[1]['id'] == 'DMA_1_2'
+        assert cs[2]['operation'] == API_DELETE
+        assert cs[2]['type'] == 'district_metering_area'
+        assert cs[2]['id'] == 'DMA_1_3'
+        assert cs[3]['operation'] == API_ADD
+        assert cs[3]['type'] == 'district_metering_area'
+        assert cs[3]['id'] == 'DMA_1_1'
+        assert cs[4]['operation'] == API_ADD
+        assert cs[4]['type'] == 'district_metering_area'
+        assert cs[4]['id'] == 'DMA_1_2'
+        assert cs[5]['operation'] == API_ADD
+        assert cs[5]['type'] == 'district_metering_area'
+        assert cs[5]['id'] == 'DMA_1_3'
 
         dmas = get_all_district_metering_area_ids(p)
         assert len(dmas) == 3
+        assert dmas[0] == 'DMA_1_1'
+        assert dmas[1] == 'DMA_1_2'
+        assert dmas[2] == 'DMA_1_3'
 
         cs = generate_sub_district_metering_area(p, 'DMA_1_1', 2).operations
         assert len(cs) == 2
+        assert cs[0]['operation'] == API_ADD
+        assert cs[0]['type'] == 'district_metering_area'
+        assert cs[0]['id'] == 'DMA_[DMA_1_1]_2_1'
+        assert cs[1]['operation'] == API_ADD
+        assert cs[1]['type'] == 'district_metering_area'
+        assert cs[1]['id'] == 'DMA_[DMA_1_1]_2_2'
+
+        cs = generate_sub_district_metering_area(p, 'DMA_1_2', 3).operations
+        assert len(cs) == 2
+        assert cs[0]['operation'] == API_ADD
+        assert cs[0]['type'] == 'district_metering_area'
+        assert cs[0]['id'] == 'DMA_[DMA_1_2]_2_2'
+        assert cs[1]['operation'] == API_ADD
+        assert cs[1]['type'] == 'district_metering_area'
+        assert cs[1]['id'] == 'DMA_[DMA_1_2]_2_3'
+
+        cs = generate_sub_district_metering_area(p, 'DMA_1_3', 2).operations
+        assert len(cs) == 2
+        assert cs[0]['operation'] == API_ADD
+        assert cs[0]['type'] == 'district_metering_area'
+        assert cs[0]['id'] == 'DMA_[DMA_1_3]_2_1'
+        assert cs[1]['operation'] == API_ADD
+        assert cs[1]['type'] == 'district_metering_area'
+        assert cs[1]['id'] == 'DMA_[DMA_1_3]_2_2'
 
         dmas = get_all_district_metering_area_ids(p)
-        assert len(dmas) == 5
+        assert len(dmas) == 9
 
         cs = generate_district_metering_area(p, 3).operations
-        assert len(cs) == 8
+        assert len(cs) == 12
 
         dmas = get_all_district_metering_area_ids(p)
         assert len(dmas) == 3