test_mycode.py

from mycode import add

def test_add():
    assert add(2, 2) == 4

def test_add_more():
    assert add(2, 3) == 5