Improve test coverage

src/traccuracy/track_errors/basic.py

@@ -30,7 +30,7 @@ def _classify_nodes(matched: Matched):
     gt_graph = matched.gt_graph
 
     if pred_graph.node_errors and gt_graph.node_errors:
-        logger.info("Node errors already calculated. Skipping graph annotation")
+        logger.warning("Node errors already calculated. Skipping graph annotation")
         return
 
     # Label as TP if the node is matched
@@ -72,7 +72,7 @@ def _classify_edges(matched: Matched):
     gt_graph = matched.gt_graph
 
     if pred_graph.edge_errors and gt_graph.edge_errors:
-        logger.info("Edge errors already calculated. Skipping graph annotation")
+        logger.warning("Edge errors already calculated. Skipping graph annotation")
         return
 
     # Node errors are needed for edge annotation

tests/track_errors/test_basic.py

@@ -23,7 +23,7 @@ def test_empty_pred(self):
         for attrs in matched.gt_graph.nodes.values():
             assert NodeFlag.FALSE_NEG in attrs
 
-    def test_good_match(self):
+    def test_good_match(self, caplog):
         matched = ex_graphs.good_matched()
         _classify_nodes(matched)
 
@@ -32,6 +32,12 @@ def test_good_match(self):
             for attrs in graph.nodes.values():
                 assert NodeFlag.TRUE_POS in attrs
 
+        # Check that it doesn't run a second time
+        _classify_nodes(matched)
+        assert (
+            "Node errors already calculated. Skipping graph annotation" in caplog.text
+        )
+
     @pytest.mark.parametrize("t", [0, 1, 2])
     def test_fn_node(self, t):
         wrong_node = [1, 2, 3][t]
@@ -130,14 +136,20 @@ def test_empty_pred(self):
         for attrs in matched.gt_graph.edges.values():
             assert EdgeFlag.FALSE_NEG in attrs
 
-    def test_good_match(self):
+    def test_good_match(self, caplog):
         matched = ex_graphs.good_matched()
         _classify_edges(matched)
 
         for graph in [matched.gt_graph, matched.pred_graph]:
             for attrs in graph.edges.values():
                 assert EdgeFlag.TRUE_POS in attrs
 
+        # Check that it doesn't run a second time
+        _classify_edges(matched)
+        assert (
+            "Edge errors already calculated. Skipping graph annotation" in caplog.text
+        )
+
     def test_fn_node_end(self):
         matched = ex_graphs.fn_node_matched(0)
         _classify_edges(matched)