How to import module into a python unit test

Question

I know this has been asked a lot of times, been looking at it for the last two hours and still can't figure it out. Would be very grateful for a solution to my actual problem rather than a link to someone else problem if possible.

Here goes...

/src
    /moduleA
        __init__.py
        messages.py
    /tests
        testA.py

messages.py

def get_messages(paramA):
    if paramA:
        return "A"
    else:
        return "B"

testA.py

import unittest
from moduleA import messages

My unit test will simply check whether A or B is returned from the get_messages(...) function.

As I understand it, moduleA is a module as it has an __init__.py file. VSCode does not complain about the from moduleA import messages line in my test however when I try to execute testA.py I get the following error message...

ImportError: No module named moduleA.

Also tried running it with python -m

Would be very grateful if someone can tell me what I'm doing wrong. I can change my file structure around if that would help. Perhaps the examples I've been reading are for Python 3 and I'm using 2.7?

Thanks.

wholevinski · Accepted Answer · 2018-08-08 10:20:48Z

5

Here's my suggestion on how to take care of this: ValueError: Attempted relative import in non-package not for tests package

So you'll need to add a __init__.py to your test directory, back up one directory to src then run python -m test.testA and you should be good to go.

Also, once you see that to run successfully, you'll see that you have a syntax error in get_messages: you're missing a colon after else.

answered Aug 8, 2018 at 10:20

wholevinski

3,85821 silver badges25 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

William Baker Morrison · Accepted Answer · 2021-01-08 17:31:03Z

-2

Firstly, we should check configuration before running the file.

Cross check with spaces like:

ENVIRONMENT=mock
PYTHONPATH=./src

edited Jan 8, 2021 at 17:31

William Baker Morrison

1,8074 gold badges22 silver badges35 bronze badges

answered Jan 8, 2021 at 9:46

user14965056

1

Collectives™ on Stack Overflow

How to import module into a python unit test

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related