Skip to content

Commit 21960e8

Browse files
fix test
1 parent c129a0e commit 21960e8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/results/test_disease_observer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -99,8 +99,8 @@ def test_previous_state_update(base_config, base_plugins, disease, model):
9999
state_cols = [observer.previous_state_column_name, observer.disease]
100100
pop0 = simulation.get_population(state_cols)
101101

102-
# Assert that the previous_state column is empty at initialization
103-
assert pop0[observer.previous_state_column_name].isna().all()
102+
# Assert that the previous_state column equals the current state column
103+
assert (pop0[observer.previous_state_column_name] == pop0[observer.disease]).all()
104104
assert pop0[observer.disease].notna().all()
105105

106106
simulation.step()

0 commit comments

Comments
 (0)