How can I mock "import pyspark" statement in python

Question

My python script with "import pyspark" import failed saying "could not find pyspark module". My code deployment is required to pass through Sonar scan and pyspark module name not listed in our enterprise level artifacts portal and hence the fail.

Can someone please help me understand how can I mock import statement to get out of this issue

thanks in advance.

I have mocked all calls to pyspark API which I am using in my code. But I am not aware of how to mock import statements

Answer 1

Finally, I got this resolved. sys.modules list helped me.

import sys

sys.modules['pyspark'] = dummy_spark_implementation()

I put this statement in my unit test script just before the import statement of my main script(the script to be tested).

this worked for me