the test script resides in another location (tests subdir) compared to main.py.

So probably what is going on is that the test script cannot find the necessary files (modules).
So make sure that either the modules are installed or your python path is set.
it is also important to check where the tests are run: locally or on a databricks cluster.