def test_dev_by_zero(self):
# arrange
c = Calculator()
a = 5
b = 0
# act
result = c.div(a, b)
# assert
self.assertEqual('На ноль делить нельзя', result)
def test_add_success(self):
# arrange
c = Calculator()
a = 2
b = 2
# act
result = c.add(a, b)
# assert
self.assertEqual(4, result)
def test_empty_b(self):
# arrange
c = Calculator()
a = 5
b = ''
# act
result = c.add(a, b)
# assert
self.assertEqual(
'2ое число содержит некорректные данные, введите число', result)
def test_str_a(self):
# arrange
c = Calculator()
a = 'x'
b = 5
# act
result = c.add(a, b)
# assert
self.assertEqual(
'1ое число содержит некорректные данные, введите число', result)
from HW.Calculator import Calculator
c = Calculator()
a = int(input("Введите 1ое число: "))
b = int(input("Введите 2ое число: "))
action = input("Выберите действие: +, -, *, /: ")
if action == "+":
c.add(a, b)
elif action == "-":
c.sub(a, b)
elif action == "*":
c.mult(a, b)
elif action == "/":
c.div(a, b)