def test_file_write(self): with mock.patch("os.path.exists") as mock_exists: mock_exists.return_value = True with mock.patch("__builtin__.open") as mock_open: mock_open.return_value = StringIO() mock_open.return_value.close = lambda: None file_operation.baba("test.txt") self.assertEqual(mock_open.return_value.getvalue(), "test")
def test_baba(self): with mock.patch("os.path.exists") as mock_exists: mock_exists.return_value = False self.assertRaises(Exception, lambda: file_operation.baba("test.txt"))