Fix index out ouf bounds for axis error

Author: stefanDeveloper

src/inspector/inspector.py

@@ -326,11 +326,20 @@ def _count_errors(
             unique_times, _, unique_counts = np.unique(
                 timestamps, return_index=True, return_counts=True
             )
-            time_indices = (
-                ((unique_times - min_date) // TIME_RANGE)
-                .astype(f"timedelta64[{TIME_TYPE}]")
-                .astype(int)
-            )
+
+            # Compute indices
+            deltas = unique_times - min_date
+            time_indices = (deltas / np.timedelta64(TIME_RANGE, TIME_TYPE)).astype(int)
+
+            # Filter out-of-range indices
+            valid_mask = (time_indices >= 0) & (time_indices < counts.size)
+            if not np.all(valid_mask):
+                invalid_count = np.count_nonzero(~valid_mask)
+                logger.warning(f"{invalid_count} timestamps outside expected time range — ignored.")
+                time_indices = time_indices[valid_mask]
+                unique_counts = unique_counts[valid_mask]
+
+            # Fill counts
             counts[time_indices] = unique_counts
         else:
             logger.warning("Empty messages to inspect.")

tests/inspector/test_inspector.py

@@ -221,6 +221,56 @@ def test_clear_data_with_existing_data(
 
 class TestDataFunction(unittest.TestCase):
 
+    @patch("src.inspector.inspector.ClickHouseKafkaSender")
+    @patch("src.inspector.inspector.ExactlyOnceKafkaProduceHandler")
+    @patch("src.inspector.inspector.ExactlyOnceKafkaConsumeHandler")
+    @patch("src.inspector.inspector.TIME_TYPE", "ms")
+    @patch("src.inspector.inspector.TIME_RANGE", 1)
+    def test_count_errors_valid_and_out_of_range(
+        self, mock_kafka_consume_handler, mock_produce_handler, mock_clickhouse
+    ):
+        """Test that _count_errors correctly counts events
+        and ignores timestamps outside the [begin, end] range.
+        """
+
+        # Mock Kafka handlers to avoid side effects
+        mock_kafka_consume_handler.return_value = MagicMock()
+        mock_produce_handler.return_value = MagicMock()
+        mock_clickhouse.return_value = MagicMock()
+
+        sut = Inspector()
+
+        # Prepare timestamps
+        begin_timestamp = datetime.now()
+        end_timestamp = begin_timestamp + timedelta(milliseconds=2)
+
+        # Message within range
+        in_range = DEFAULT_DATA.copy()
+        in_range["timestamp"] = datetime.strftime(
+            begin_timestamp + timedelta(milliseconds=1),
+            TIMESTAMP_FORMAT,
+        )
+
+        # Message outside range (should be ignored)
+        out_of_range = DEFAULT_DATA.copy()
+        out_of_range["timestamp"] = datetime.strftime(
+            begin_timestamp + timedelta(milliseconds=10),
+            TIMESTAMP_FORMAT,
+        )
+
+        messages = [in_range, out_of_range]
+
+        # Execute
+        result = sut._count_errors(messages, begin_timestamp, end_timestamp)
+
+        # The expected counts:
+        # 1 ms → one event
+        # 0 ms, 2 ms → no events
+        expected = np.asarray([[0.0], [1.0], [0.0]])
+
+        np.testing.assert_array_equal(result, expected)
+
+
     @patch("src.inspector.inspector.ClickHouseKafkaSender")
     @patch("src.inspector.inspector.ExactlyOnceKafkaProduceHandler")
     @patch("src.inspector.inspector.ExactlyOnceKafkaConsumeHandler")