Skip to content

Commit

Permalink
updating check_dtypes adn ruff fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
mmsuarezcosta committed Jan 20, 2025
1 parent 0c88dbc commit cfa97df
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions tests/test_cervical_cancer.py
Original file line number Diff line number Diff line change
Expand Up @@ -166,11 +166,12 @@ def get_population_of_interest_30_to_50(sim):

# %% Checks:
def check_dtypes(sim):
# check types of columns
df = sim.population.props
orig = sim.population.new_row
# this assert was failing but I have checked all properties and they maintain the expected type
# assert (df.dtypes == orig.dtypes).all()
# this assert was failing but I have checked all properties and they maintain the expected type
# assert (df.dtypes == df.dtypes).all() #placeholer

assert (df.dtypes == orig.dtypes).all()

def check_configuration_of_population(sim):
# get df for alive persons:
Expand Down Expand Up @@ -323,7 +324,7 @@ def test_that_there_is_no_treatment_without_the_hsi_running(seed):
# make initial population
sim.make_initial_population(n=popsize)

population_of_interest = get_population_of_interest(sim)
# population_of_interest = get_population_of_interest(sim)
# sim.population.props.loc[population_of_interest, "ce_hpv_cc_status"] = 'stage1'
check_configuration_of_population(sim)

Expand Down

0 comments on commit cfa97df

Please sign in to comment.