Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 10 additions & 1 deletion src/pyrecest/filters/measurement_scoring.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

from __future__ import annotations

from collections.abc import Sequence
from dataclasses import dataclass
from typing import Any

Expand All @@ -25,11 +26,19 @@ class MeasurementScore:
predicted_measurements: Any | None
innovation_covariance: Any | None
residual: Any | None
active_measurement_indices: list[int]
active_measurement_indices: Sequence[int]
measurement_weights: Any | None = None
quadratic_form: float | None = None
skipped_reason: str | None = None

def __post_init__(self) -> None:
# Own caller-provided sequences so this frozen result cannot be mutated indirectly.
object.__setattr__(
self,
"active_measurement_indices",
tuple(self.active_measurement_indices),
)

@property
def is_active(self) -> bool:
"""Return whether the score contains at least one active measurement."""
Expand Down
10 changes: 10 additions & 0 deletions tests/filters/test_measurement_scoring.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,3 +15,13 @@ def _score(active_measurement_indices):
def test_measurement_score_is_active_handles_numpy_index_arrays():
assert _score(np.array([0, 2])).is_active
assert not _score(np.array([], dtype=int)).is_active


def test_measurement_score_owns_active_measurement_indices():
active_measurement_indices = []
score = _score(active_measurement_indices)

active_measurement_indices.append(0)

assert score.active_measurement_indices == ()
assert not score.is_active
Loading