class TestMain(unittest.TestCase): def setUp(self): self.pm = Base() def test_remove_searching(self): data = self.pm.remove_searching(['SEARCHING...']) self.assertEqual(data, []) data = self.pm.remove_searching(['SEARCHING...', '410000000000']) self.assertEqual(data, ['410000000000']) # Exception - ValueError data = self.pm.remove_searching(['410C34']) self.assertEqual(data, ['410C34']) def test_create_data(self): with self.assertRaises(NotImplementedError) as cm: self.pm.create_data([]) self.assertIsInstance(cm.exception, NotImplementedError) def test_check_result(self): resp = self.pm.check_result(['NODATA']) self.assertFalse(resp) resp = self.pm.check_result(['410C7D']) self.assertTrue(resp) @mock.patch('sys.stdout') def test_check_error(self, mock_out): # if hasn't error resp = self.pm.check_error(['410C7D']) self.assertTrue(resp) # if some error appears resp = self.pm.check_error(['7F0112']) self.assertFalse(resp) resp = self.pm.check_error(['7F0188']) self.assertFalse(resp)