pytest - импортирование пропущенного атрибута класса

Question

Я новичок в python и пытаюсь написать простой тест для проверки класса.

Моя структура папок выглядит следующим образом:

lib
  mycode
     __init__.py
     my_calc.py
  tests
     __init__.py
     test_my_calc.py

Я запускаю Pytest внутри папка lib.

Вот мой python класс:

class MyCalc:

    def add(a,b):
        return a + b

И мой test_my_cal c .py выглядит так:

from mycode import my_calc

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

Я получаю эта ошибка при запуске pytest:

E       AttributeError: module 'mycode.my_calc' has no attribute 'add'

Итак, я попытался импортировать бит Class, получил другую ошибку:

from mycode import MyCalc

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

Я получаю эту ошибку:

E   ImportError: cannot import name 'MyCalc' from 'mycode'

Есть ли что-то очевидное, чего мне не хватает?