import pytest
from poetry_template.main import sum

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)