test_math_integrations.py

import pytest
from math_operations import add, substract, multiply, mean


def test_add():
    assert add(2, 3) == 5
    assert add(-1, 1) == 0


@pytest.mark.parametrize(
    "a, b, expected", [(2, 3, 5), (-1, 1, 0), (0, 0, 0), (1.5, 2.5, 4)]
)
def test_add(a, b, expected):
    assert add(a, b) == expected


def test_substract():
    assert substract(10, 5) == 5
    assert substract(-1, -1) == 0


def test_multiply():
    assert multiply(3, 5) == 15
    assert multiply(-1, -1) == 1

@pytest.mark.parametrize("numbers, expected", [
  ([10, 20, 30], 20),
  ([1.5,2.5,3.5], 2.5)
 ])

def test_mean(numbers, expected):
    if numbers:
        assert mean(numbers) == expected