From 7585748ad43306223438c080d6096d310d584448 Mon Sep 17 00:00:00 2001
From: dilpath <dilan.private+github@outlook.com>
Date: Tue, 3 Dec 2024 14:17:18 +0100
Subject: [PATCH] test

---
 test/analyze/input/models.yaml | 66 ++++++++++++++++++++++++++++
 test/analyze/test_analyze.py   | 80 ++++++++++++++++++++++++++++++++++
 2 files changed, 146 insertions(+)
 create mode 100644 test/analyze/input/models.yaml
 create mode 100644 test/analyze/test_analyze.py

diff --git a/test/analyze/input/models.yaml b/test/analyze/input/models.yaml
new file mode 100644
index 0000000..264e115
--- /dev/null
+++ b/test/analyze/input/models.yaml
@@ -0,0 +1,66 @@
+- criteria:
+    AIC: 5
+  model_id: model_1
+  model_subspace_id: M
+  model_subspace_indices:
+  - 0
+  - 1
+  - 1
+  iteration: 1
+  parameters:
+    k1: 0.2
+    k2: estimate
+    k3: estimate
+  estimated_parameters:
+    k2: 0.15
+    k3: 0.0
+  petab_yaml: ../../../doc/examples/model_selection/petab_problem.yaml
+  predecessor_model_hash: dummy_p0-0
+- criteria:
+    AIC: 4
+  model_id: model_2
+  model_subspace_id: M
+  model_subspace_indices:
+  - 1
+  - 1
+  - 0
+  iteration: 5
+  parameters:
+    k1: estimate
+    k2: estimate
+    k3: 0
+  petab_yaml: ../../../doc/examples/model_selection/petab_problem.yaml
+  predecessor_model_hash: virtual_initial_model-
+- criteria:
+    AIC: 3
+  model_id: model_3
+  model_subspace_id: M2
+  model_subspace_indices:
+  - 0
+  - 1
+  - 1
+  iteration: 1
+  parameters:
+    k1: 0.2
+    k2: estimate
+    k3: estimate
+  estimated_parameters:
+    k2: 0.15
+    k3: 0.0
+  petab_yaml: ../../../doc/examples/model_selection/petab_problem.yaml
+  predecessor_model_hash: virtual_initial_model-
+- criteria:
+    AIC: 2
+  model_id: model_4
+  model_subspace_id: M2
+  model_subspace_indices:
+  - 1
+  - 1
+  - 0
+  iteration: 2
+  parameters:
+    k1: estimate
+    k2: estimate
+    k3: 0
+  petab_yaml: ../../../doc/examples/model_selection/petab_problem.yaml
+  predecessor_model_hash: virtual_initial_model-
diff --git a/test/analyze/test_analyze.py b/test/analyze/test_analyze.py
new file mode 100644
index 0000000..2b08459
--- /dev/null
+++ b/test/analyze/test_analyze.py
@@ -0,0 +1,80 @@
+from pathlib import Path
+
+import pytest
+
+from petab_select import (
+    VIRTUAL_INITIAL_MODEL,
+    Criterion,
+    ModelHash,
+    Models,
+    analyze,
+)
+
+base_dir = Path(__file__).parent
+
+DUMMY_HASH = "dummy_p0-0"
+VIRTUAL_HASH = ModelHash.from_hash(VIRTUAL_INITIAL_MODEL)
+
+
+@pytest.fixture
+def models() -> Models:
+    return Models.from_yaml(base_dir / "input" / "models.yaml")
+
+
+def test_group_by_predecessor_model(models: Models) -> None:
+    """Test ``analyze.group_by_predecessor_model``."""
+    groups = analyze.group_by_predecessor_model(models)
+    # Expected groups
+    assert len(groups) == 2
+    assert VIRTUAL_HASH in groups
+    assert DUMMY_HASH in groups
+    # Expected group members
+    assert len(groups[DUMMY_HASH]) == 1
+    assert "M-011" in groups[DUMMY_HASH]
+    assert len(groups[VIRTUAL_HASH]) == 3
+    assert "M-110" in groups[VIRTUAL_HASH]
+    assert "M2-011" in groups[VIRTUAL_HASH]
+    assert "M2-110" in groups[VIRTUAL_HASH]
+
+
+def test_group_by_iteration(models: Models) -> None:
+    """Test ``analyze.group_by_iteration``."""
+    groups = analyze.group_by_iteration(models)
+    # Expected groups
+    assert len(groups) == 3
+    assert 1 in groups
+    assert 2 in groups
+    assert 5 in groups
+    # Expected group members
+    assert len(groups[1]) == 2
+    assert "M-011" in groups[1]
+    assert "M2-011" in groups[1]
+    assert len(groups[2]) == 1
+    assert "M2-110" in groups[2]
+    assert len(groups[5]) == 1
+    assert "M-110" in groups[5]
+
+
+def test_get_best_by_iteration(models: Models) -> None:
+    """Test ``analyze.get_best_by_iteration``."""
+    groups = analyze.get_best_by_iteration(models, criterion=Criterion.AIC)
+    # Expected groups
+    assert len(groups) == 3
+    assert 1 in groups
+    assert 2 in groups
+    assert 5 in groups
+    # Expected best models
+    assert groups[1].get_hash() == "M2-011"
+    assert groups[2].get_hash() == "M2-110"
+    assert groups[5].get_hash() == "M-110"
+
+
+def test_get_relative_criterion_values(models: Models) -> None:
+    """Test ``analyze.get_relative_criterion_values``."""
+    criterion_values = [model.get_criterion(Criterion.AIC) for model in models]
+    test_value = analyze.get_relative_criterion_values(criterion_values)
+    expected_value = [
+        criterion_value - min(criterion_values)
+        for criterion_value in criterion_values
+    ]
+    assert test_value == expected_value