Skip to content

Commit cfa97df

Browse files
committed
updating check_dtypes adn ruff fixes
1 parent 0c88dbc commit cfa97df

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/test_cervical_cancer.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -166,11 +166,12 @@ def get_population_of_interest_30_to_50(sim):
166166

167167
# %% Checks:
168168
def check_dtypes(sim):
169-
# check types of columns
170169
df = sim.population.props
171170
orig = sim.population.new_row
172-
# this assert was failing but I have checked all properties and they maintain the expected type
173-
# assert (df.dtypes == orig.dtypes).all()
171+
# this assert was failing but I have checked all properties and they maintain the expected type
172+
# assert (df.dtypes == df.dtypes).all() #placeholer
173+
174+
assert (df.dtypes == orig.dtypes).all()
174175

175176
def check_configuration_of_population(sim):
176177
# get df for alive persons:
@@ -323,7 +324,7 @@ def test_that_there_is_no_treatment_without_the_hsi_running(seed):
323324
# make initial population
324325
sim.make_initial_population(n=popsize)
325326

326-
population_of_interest = get_population_of_interest(sim)
327+
# population_of_interest = get_population_of_interest(sim)
327328
# sim.population.props.loc[population_of_interest, "ce_hpv_cc_status"] = 'stage1'
328329
check_configuration_of_population(sim)
329330

0 commit comments

Comments
 (0)