Skip to content

Commit

Permalink
small new tests
Browse files Browse the repository at this point in the history
  • Loading branch information
tjlane committed Jan 26, 2025
1 parent 682cb76 commit 310ccd7
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions test/unit/test_validate.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
from pathlib import Path

import numpy as np
from numpy.testing import assert_almost_equal

Expand Down Expand Up @@ -37,6 +39,27 @@ def test_map_negentropy(noise_free_map: Map, very_noisy_map: Map) -> None:
assert validate.map_negentropy(noise_free_map) > validate.map_negentropy(very_noisy_map)


def test_maximizer_metadata_smoke(tv_denoise_result_source_data: dict) -> None:
validate.MaximizerScanMetadata(**tv_denoise_result_source_data)


def test_maximizer_metadata_json_roundtrip(tv_denoise_result_source_data: dict) -> None:
metadata = validate.MaximizerScanMetadata(**tv_denoise_result_source_data)
json_metadata = metadata.json()
new_metadata = metadata.from_json(json_metadata)
assert new_metadata == metadata


def test_maximizer_metadata_read_write_roundtrip(
tv_denoise_result_source_data: dict, tmp_path: Path
) -> None:
metadata = validate.MaximizerScanMetadata(**tv_denoise_result_source_data)
json_file = tmp_path / "metadata.json"
metadata.to_json_file(json_file)
new_metadata = metadata.from_json_file(json_file)
assert new_metadata == metadata


def test_negentropy_maximizer_explicit() -> None:
maximizer = validate.ScalarMaximizer(objective=parabolic_objective)
test_values = np.linspace(-5, 5, 11)
Expand Down

0 comments on commit 310ccd7

Please sign in to comment.