Procházet zdrojové kódy

add test for initial state dict (#3727)

benedikt-bartscher před 9 měsíci
rodič
revize
129adc941a
1 změnil soubory, kde provedl 15 přidání a 0 odebrání
  1. 15 0
      tests/test_state.py

+ 15 - 0
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: