diff --git a/source/test/common/test_data_ports.py b/source/test/common/test_data_ports.py
index <HASH>..<HASH> 100644
--- a/source/test/common/test_data_ports.py
+++ b/source/test/common/test_data_ports.py
@@ -87,6 +87,9 @@ def test_unique_port_names():
     state.add_output_data_port("in", "int", 0)
     state.add_input_data_port("out", "int", 0)
 
+    assert len(state.input_data_ports) == 2
+    assert len(state.output_data_ports) == 2
+
     state = HierarchyState('hierarchy state')
 
     state.add_input_data_port("in", "int", 0)
@@ -115,6 +118,10 @@ def test_unique_port_names():
     state.add_input_data_port("scope", "int", 0)
     state.add_output_data_port("scope", "int", 0)
 
+    assert len(state.input_data_ports) == 3
+    assert len(state.output_data_ports) == 3
+    assert len(state.scoped_variables) == 3
+
 
 if __name__ == '__main__':
     test_default_values_of_data_ports()
Extend tests to assert correct port count