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)