示例#1
0
def test_infection_color(input_data, expected_output):
    assert expected_output in infection_color(input_data)
示例#2
0
 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)