def test_has_name_in_list(): c = Compartment("infected") assert c.has_name_in_list(["infected", "foo", "bar"]) assert c.has_name_in_list( [Compartment("infected"), Compartment("foo"), Compartment("bar")]) assert not c.has_name_in_list(["infected_stuff", "foo", "bar"]) assert not c.has_name_in_list([ Compartment("infected_stuff"), Compartment("foo"), Compartment("bar") ])
def test_has_name_in_list_stratified(): c = Compartment( "infected", strat_names=["location", "age"], strat_values={ "location": "hawaii", "age": "15" }, ) assert c.has_name_in_list(["infected", "foo", "bar"]) assert c.has_name_in_list( [Compartment("infected"), Compartment("foo"), Compartment("bar")]) assert not c.has_name_in_list(["infected_stuff", "foo", "bar"]) assert not c.has_name_in_list([ Compartment("infected_stuff"), Compartment("foo"), Compartment("bar") ])