如何测试python数据库交互功能

Question

我的python文件course.py中有一些函数，它位于proj / com / api / course中 夹。这些功能与数据库交互。

我正在使用烧瓶，sqlalchemy。

course.py

def create(**kwargs):
            user = User()
            session = Session()
            user.name = kwargs['name']
            user.email = kwargs['email']
            session.add(user)
            session.commit()
            session.close()
            return user
def get_data_by_id(user_id):
        session = Session()
        user = session.query(User).filter(User.id==user_id).first()
        session.close()
        return user

我的问题是我应该在哪里创建测试模块以及如何设置与数据库的连接以测试函数？