test_calculator.py

"""
Unit tests for the calculator library
"""


import calculator


class TestCalculator:

    def test_addition(self):
        assert 4 == calculator.add(2, 2)

    def test_subtraction(self):
        assert 2 == calculator.subtract(4, 2)

    def test_multiplication(self):
        assert 42 == calculator.multiply(7, 6)

    def test_division(self):
        assert 7 == calculator.divide(42, 6)