From 129adc941a6bb2ae5a9de6c010144c17a93823ee Mon Sep 17 00:00:00 2001
From: benedikt-bartscher
 <31854409+benedikt-bartscher@users.noreply.github.com>
Date: Wed, 31 Jul 2024 20:37:53 +0200
Subject: [PATCH] add test for initial state dict (#3727)

---
 tests/test_state.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/tests/test_state.py b/tests/test_state.py
index 18d740015..aa5705b09 100644
--- a/tests/test_state.py
+++ b/tests/test_state.py
@@ -1012,6 +1012,21 @@ def interdependent_state() -> BaseState:
     return s
 
 
+def test_interdependent_state_initial_dict() -> None:
+    s = InterdependentState()
+    state_name = s.get_name()
+    d = s.dict(initial=True)[state_name]
+    d.pop("router")
+    assert d == {
+        "x": 0,
+        "v1": 0,
+        "v1x2": 0,
+        "v2x2": 2,
+        "v1x2x2": 0,
+        "v3x2": 2,
+    }
+
+
 def test_not_dirty_computed_var_from_var(
     interdependent_state: InterdependentState,
 ) -> None: