def test_save_as_foo(self): image = Image() image.save_as_jpeg = mock.MagicMock() with self.assertRaises(ValueError): image.save("foo", "outfile") self.assertFalse(image.save_as_jpeg.mock_calls)
def test_save_as_jpeg(self): image = Image() image.save_as_jpeg = mock.MagicMock() image.save("jpeg", "outfile") image.save_as_jpeg.assert_called_with("outfile")