Unittest Passing arguments from another script

Question

So here is a pretty simple script im running from an webservice:

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import sys

test=int(sys.argv[1])+int(sys.argv[2])
print(test)

and now I want to write an Unittest script to test this script, but I dont know how I should pass the 2 required arguments inside a unittest function. Here is my current code:

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import unittest
import ExecuteExternal

class TestUnittest(unittest.TestCase):

    def test(self):
       self.assertTrue(ExecuteExternal, "5", "5"==10)

if __name__ == '__main__':
    unittest.main()

I did search here in the forum but was not able to find an answer

M.Rau · Accepted Answer · 2018-08-13 08:14:03Z

1

You can wrangle the sys args but the preferred way is to wrap your logic in a function, and put the typical main check around the function call in your script.

Them you can import the script in your unit test and call the function instead of ruining the script.

This is an untested version of your script:

# -*- coding: utf-8 -*-

import sys

def calc(a, b):
    return int(a)+int(b)

if __name__ == "__main__":
    print(calc(sys.argv[1], sys.argv[2]))

assuming your script is called ExecuteExternal.py then your test script goes:

# -*- coding: utf-8 -*-

import unittest
import ExecuteExternal

class TestUnittest(unittest.TestCase):

    def test(self):
        result = ExecuteExternal.calc("5", "6")
        self.assertTrue(result, 11)

if __name__ == '__main__':
    unittest.main()

edited Aug 13, 2018 at 8:14

answered Aug 13, 2018 at 6:39

M.Rau

1,1147 silver badges10 bronze badges

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

1 Comment

Kevin Müller Over a year ago

My main problem is that the script is called by my webservice with some arguments and if I include the unittest within the script itselve unittest gives me 2 errors which are like: module "main" has no attribute: "argument1" and the same for the second argument

M.Rau · Accepted Answer · 2018-08-13 07:59:03Z

0

Furthermore you should watch your naming convention for the script. See PEP8 at https://www.python.org/dev/peps/pep-0008/#package-and-module-names

answered Aug 13, 2018 at 7:59

M.Rau

1,1147 silver badges10 bronze badges

Collectives™ on Stack Overflow

Unittest Passing arguments from another script

2 Answers 2

1 Comment

Comments

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Related