def test_data3():
code_result = set(meetup(data3))
true_result = set([
Agent("Mark", Type.DYING),
Agent("Mork", Type.HEALTHY),
Agent("Harry", Type.DEAD),
Agent("Cure", Type.CURE),
Agent("Lora", Type.HEALTHY),
Agent("Monica", Type.SICK),
])
assert code_result == true_result
def test_data2():
code_result = set(meetup(data2))
true_result = set([
Agent("Zelda0", Type.DYING),
Agent("Zelda1", Type.DYING),
Agent("Zelda2", Type.DYING),
Agent("Zelda3", Type.DYING),
Agent("Zelda4", Type.DEAD),
Agent("Zelda5", Type.HEALTHY),
])
assert code_result == true_result
def test_data0():
code_result = set(meetup(data0))
true_result = {
Agent(name="Adam", category=Type.HEALTHY),
Agent(name="Alice", category=Type.DEAD),
Agent(name="Bob", category=Type.HEALTHY),
Agent(name="Charlie", category=Type.SICK),
Agent(name="Cure0", category=Type.CURE),
Agent(name="Cure1", category=Type.CURE),
Agent(name="Cure2", category=Type.CURE),
Agent(name="Darlene", category=Type.DEAD),
Agent(name="Emma", category=Type.HEALTHY),
Agent(name="Vaccine", category=Type.DYING),
}
assert code_result == true_result
def test_data5():
code_result = meetup(data5)
true_result = []
assert code_result == true_result
def test_data4():
code_result = set(meetup(data4))
true_result = {Agent("Robert", Type.SICK)}
assert code_result == true_result
def test_data1():
code_result = set(meetup(data1))
true_result = set(data1)
assert code_result == true_result
def test_data6():
code_result = meetup(data5)
assert isinstance(code_result, list)