def test_ProperOutput(self): ''' Checks if output is as expected. ''' input_str = "SampleNumber=3234 provider=Dr. M. Welby patient=John Smith priority=High" output = CommandLine.CommandLine(input_str) self.assertEqual(output, "12=4 8=12 7=10 8=4")
def test_WrongInput(self): ''' Checks if output is equal -1 for wrong input. ''' output = CommandLine.CommandLine(12) self.assertEqual(output, -1)