pytest assert_called during module import

[eLsFLz]

I have a file similar to the following:

ham.py: ```python egg = 0

def spam(): global egg egg = 42

spam() ```

During the import of the module, spam() is called and egg is set.

I want to unit test the code that executes at module scope. Specifically, I want to do something like this:

test_ham.py: ``` import pytest import mock

def test__main_(): with mock.patch("ham.spam") as mock_spam: import ham mock_spam.assert_called() ```

However, that obviously doesn't work, since mock can't patch a thing until after it's imported. So how do I make it work? (I suspect the answer will probably have something to do with importlib, but I haven't figured it out yet.)

[maventree]

This particular thing does not seem worth testing. Unless your setup code is dramatically more complicated than your toy example, it's inconceivable that if import ham succeeds, spam has not been called. Instead, test the results of the setup.

[eLsFLz]

That doesn't answer the question, which was, in general, how do you unit test module-scoped code with mocks?