python-template/tests/test_main.py

import pytest
from python_template.main import sum, mul

def test_sum():
    a = 1
    b = 2
    s = 3

    assert s == sum(a, b)


def test_sum_wrong():
    a = 1
    b = 2
    s_wrong = 4

    assert s_wrong != sum(a, b)

def test_mul():
    a = 1
    b = 2
    m = 2

    assert m == mul(a, b)