def test_infection_color(input_data, expected_output): assert expected_output in infection_color(input_data)
def test_infection_color(self): assert 'color:green' in infection_color('clean') assert 'color:green' in infection_color(0) assert 'color:red' in infection_color('foo') assert 'color:red' in infection_color(9999) assert 'color:red' in infection_color(None)