def test_basic(self):
demo = Demo(1, 2)
self.assertEqual(1, demo.num1)
self.assertEqual(2, demo.num2)
self.assertEqual(1, demo._special_number())
self.assertEqual(4, demo.sum())
self.assertEqual(8, demo.sum_bias())
def test_raise_error_side_effect(self):
demo = Demo(1, 2)
demo._special_number = MagicMock(side_effect=SpecialError)
with self.assertRaises(SpecialError):
demo.sum()
demo._special_number.assert_called_with()
def test_raise_error(self):
demo = Demo(1, 2)
demo._special_number = MagicMock(return_value=-1)
with self.assertRaises(ValueError):
demo.sum()
demo._special_number.assert_called_with()