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_patch(self): with patch.object(Demo, '_special_number', return_value=-1) as mocked_mathod: demo = Demo(1, 2) with self.assertRaises(ValueError): demo.sum() mocked_mathod.assert_called_once_with()
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()