import unittest from physik import do_something class MyTestCase(unittest.TestCase): def test_something(self): arg = 123456 result = do_something(arg) self.assertEqual(arg, result) if __name__ == '__main__': unittest.main()