Python Forum

Hey guys

I try to reference the class Database in my unit test.
This is the first time I program with python so I hope I explain everything correctly :-)

I have a following solution structure:

Quote:> main
> main/src/myApi.py
> main/unittests/myApiTests.py

in myApiTests.py I try to import the class Database from myApi.py
but i get the following error every time I try:

Quote:>Traceback (most recent call last):
File "S:\...\main\UnitTests\myApiTests.py", line 3, in <module>
from myApi import Database
ModuleNotFoundError: No module named 'myApi'

import sys
    sys.path.append("../src")
    from myApi import Database
    import unittest


    class BotApiTests(unittest.TestCase):
        def test_GetUserBalance_WhenUserDoesNotExist_ThenReturn0(self):
            testDatabase = Database('testDb.db')
            userName = "testUserName"

            result = testDatabase.GetUserBalance(userName)
            self.assertEqual(result, 0)

    unittest.main()

I've also tried

sys.path.append("..src")

or

   sys.path.append("..")

or

    from myApi import Database

but nothing works :-(

I have an empty __init__.py in every folder

Full Code:

main/src/myApi.py:

    import re
    import string
    import urllib.request
    import sqlite3
    import praw
    
    class Database:   
        def __init__(self, name='cryptotipbot.db'):
            print("in __init__ -> ", (name))
            self.connection = sqlite3.connect(name, check_same_thread=False)
            self.database = self.connection.cursor()
            self.CreateDatabase()
            self.addressIndex = len(self.database.execute("SELECT * FROM usedAdresses").fetchall())
    
        def CreateDatabase(self):
            print("in CreateDatabase")
            self.database.execute("CREATE TABLE IF NOT EXISTS Users (redditUsername TEXT PRIMARY KEY, balance INTEGER)")
            self.connection.commit()
            self.database.execute("CREATE TABLE IF NOT EXISTS CommentsRepliedTo (commentId TEXT PRIMARY KEY)")
            self.connection.commit()
            self.database.execute("CREATE TABLE IF NOT EXISTS UsedAdresses (adressIndex INTEGER PRIMARY KEY, adress TEXT)")
            self.connection.commit()
            self.database.execute("CREATE TABLE IF NOT EXISTS DepositRequests (messageId TEXT PRIMARY KEY, adress TEXT, amount INTEGER)")
            self.connection.commit()
        
        def CreateUser(self, redditUsername):
            user = self.database.execute("SELECT * FROM Users WHERE redditUsername = ?", (redditUsername,)).fetchone()
            if not user:
                self.database.execute("INSERT INTO Users (redditUsername, balance) VALUES (?, ?)", (redditUsername, 0))
                self.connection.commit()
    
        def GetUserBalance(self, redditUsername):
            entry = self.database.execute("SELECT * FROM Users WHERE redditUsername = ?",(redditUsername,)).fetchone()
            if entry:
                balance = entry[1]
                return balance
            else:
                self.CreateUser(redditUsername)
                return self.GetUserBalance(redditUsername)

main/UnitTests/myApiTests.py

    import unittest
    from ..src import myApi
    
    class BotApiTests(unittest.TestCase):
        def test_GetUserBalance_WhenUserDoesNotExist_ThenReturn0(self):
            testDatabase = Database('testDb.db')
            userName = "testUserName"
    
            result = testDatabase.GetUserBalance(userName)
            self.assertEqual(result, 0)
    
    
    unittest.main()

You need to provide __init__.py at top level, and
empty __init__.py in src and sub-folders.
See: https://docs.python.org/3/tutorial/modul...l#packages
The top level __init__.py lets python know where dependencies are located.

Hmm I already tried to have an empty __init__.py in every folder but that didn't work...

Does the __init__.py at top level have to have some content?

Read the docs.  See link in previous post.
The one at the top describes all of the dependencies.
It is not empty.