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")
    ])