add test to ensure we can select correct line

Author: patricktnast

tests/conftest.py

@@ -94,6 +94,8 @@ def sample_stats_file(tmp_path) -> Path:
      2459    0.200    0.000    0.800    0.000 /vivarium/framework/population/population_view.py:133(get)
    50/25    0.150    0.003    0.600    0.024 /some/custom/module.py:100(custom_function)
       100    0.025    0.000    0.500    0.005 /another/module.py:200(another_function)
+      150    0.030    0.000    0.450    0.003 /vivarium/framework/values/pipeline.py:66(duplicate_func)
+      200    0.040    0.000    0.800    0.004 /vivarium/framework/values/pipeline.py:150(duplicate_func)
 """
     stats_file = tmp_path / "test_stats.txt"
     stats_file.write_text(stats_content)

tests/test_extraction.py

@@ -224,3 +224,59 @@ def test_extract_metrics_missing_patterns(self, sample_stats_file):
         metrics = config.extract_metrics(sample_stats_file)
 
         assert metrics["missing_cumtime"] is None
+
+    def test_extract_metrics_duplicate_func_no_line_number(self, sample_stats_file):
+        """Test that we match the first occurrence when line_number is None."""
+        patterns = [
+            FunctionCallConfiguration(
+                "duplicate_first",
+                "vivarium/framework/values/pipeline.py",
+                "duplicate_func",
+                line_number=None,  # Should match first occurrence
+                extract_cumtime=True,
+                extract_percall=True,
+                extract_ncalls=True,
+            )
+        ]
+        config = ExtractionConfig(patterns=patterns)
+        metrics = config.extract_metrics(sample_stats_file)
+
+        # Should match line 66, the first occurrence
+        assert metrics["duplicate_first_cumtime"] == 0.450
+        assert metrics["duplicate_first_percall"] == 0.003
+        assert metrics["duplicate_first_ncalls"] == 150
+
+    def test_extract_metrics_duplicate_func_with_line_number(self, sample_stats_file):
+        """Test that we can match specific occurrences by line number."""
+        patterns = [
+            FunctionCallConfiguration(
+                "duplicate_line_66",
+                "vivarium/framework/values/pipeline.py",
+                "duplicate_func",
+                line_number=66,
+                extract_cumtime=True,
+                extract_percall=True,
+                extract_ncalls=True,
+            ),
+            FunctionCallConfiguration(
+                "duplicate_line_150",
+                "vivarium/framework/values/pipeline.py",
+                "duplicate_func",
+                line_number=150,
+                extract_cumtime=True,
+                extract_percall=True,
+                extract_ncalls=True,
+            ),
+        ]
+        config = ExtractionConfig(patterns=patterns)
+        metrics = config.extract_metrics(sample_stats_file)
+
+        # Line 66
+        assert metrics["duplicate_line_66_cumtime"] == 0.450
+        assert metrics["duplicate_line_66_percall"] == 0.003
+        assert metrics["duplicate_line_66_ncalls"] == 150
+
+        # Line 150
+        assert metrics["duplicate_line_150_cumtime"] == 0.800
+        assert metrics["duplicate_line_150_percall"] == 0.004
+        assert metrics["duplicate_line_150_ncalls"] == 200