Importing a Python script that requires system arguments into unit test

Question

I have a Python file, let's call it script1.py. I am trying to write a unit test (using unittest) called script1_test.py. script1 is meant to be called from command line and take in a number of arguments. When script 1 is run, it starts off with:

if __name__ == "__main__" and len(sys.argv) == 6:
    func1()
else
   print "Wrong number of arguments"
   sys.exit(1)

I'm just trying to execute and test a function (here called func1) within script1 independent of the main body of the code. But when I do so, I keep hitting the sys.exit from main during the import phase. How can I run the test without hitting this error?

maxymoo · Accepted Answer · 2016-07-22 04:11:47Z

1

When you do the import of your script, __name__ is not equal to main so you're calling the else block. Instead you should nest your if blocks:

if __name__ == "__main__":
    if len(sys.argv) == 6:
       func1()
    else:
       print "Wrong number of arguments"
       sys.exit(1)

answered Jul 22, 2016 at 4:11

maxymoo

36.7k12 gold badges97 silver badges121 bronze badges

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

Collectives™ on Stack Overflow

Importing a Python script that requires system arguments into unit test

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related