From 6ae03191c5173d1e6be4879244facb47d7d31b21 Mon Sep 17 00:00:00 2001
From: "claude[bot]" <41898282+claude[bot]@users.noreply.github.com>
Date: Wed, 21 Oct 2026 13:34:38 +0000
Subject: [PATCH] Add unit tests for rupture propagation probability helpers

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
---
 tests/test_rupture_propagation.py | 52 +++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/tests/test_rupture_propagation.py b/tests/test_rupture_propagation.py
index 7ab6cf5..3263e30 100644
--- a/tests/test_rupture_propagation.py
+++ b/tests/test_rupture_propagation.py
@@ -636,3 +636,55 @@ def test_jump_points_from_rupture_tree_min_depth(
     # shallow as the minimum depth allows.
     assert from_depth_m == pytest.approx(expected_jump_depth_m, abs=1)
     assert to_depth_m == pytest.approx(expected_jump_depth_m, abs=1)
+
+
+def test_shaw_dieterich_distance_model_at_delta():
+    assert rupture_propagation.shaw_dieterich_distance_model(1.0, 3, 1) == 1
+
+
+def test_shaw_dieterich_distance_model_decay():
+    assert rupture_propagation.shaw_dieterich_distance_model(
+        4.0, 3, 1
+    ) == pytest.approx(np.exp(-1))
+
+
+def test_shaw_dieterich_distance_model_clipped_below_delta():
+    assert rupture_propagation.shaw_dieterich_distance_model(0.0, 3, 1) == 1
+
+
+def test_prune_distance_graph_removes_edges_beyond_cutoff():
+    assert rupture_propagation.prune_distance_graph(
+        {"a": {"b": 10.0, "c": 20.0}}, cutoff=15.0
+    ) == {"a": {"b": 10.0}}
+
+
+def test_prune_distance_graph_cutoff_is_strict():
+    assert rupture_propagation.prune_distance_graph(
+        {"a": {"b": 15.0}}, cutoff=15.0
+    ) == {"a": {}}
+
+
+def test_prune_distance_graph_keeps_isolated_faults():
+    assert rupture_propagation.prune_distance_graph(
+        {"a": {"b": 20.0}, "b": {"a": 20.0}}, cutoff=15.0
+    ) == {"a": {}, "b": {}}
+
+
+def test_probability_graph_converts_metres_to_kilometres():
+    graph = rupture_propagation.probability_graph(
+        {"a": {"b": 4000.0}, "b": {"a": 4000.0}}
+    )
+    assert graph["a"]["b"]["weight"] == pytest.approx(np.exp(-1))
+
+
+def test_probability_graph_caps_weight():
+    graph = rupture_propagation.probability_graph({"a": {"b": 0.0}, "b": {"a": 0.0}})
+    assert graph["a"]["b"]["weight"] == pytest.approx(0.99)
+
+
+def test_probability_graph_keeps_isolated_nodes():
+    graph = rupture_propagation.probability_graph(
+        {"a": {"b": 4000.0}, "b": {"a": 4000.0}, "c": {}}
+    )
+    assert "c" in graph.nodes
+    assert graph.degree("c") == 0