update tests

This commit is contained in:
Akash Mahanty 2020-10-02 23:10:48 +05:30 committed by GitHub
parent a8ce970ca0
commit ede251afb3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -90,7 +90,7 @@ def test_args_handler():
args = argparse.Namespace(url=None, version=False) args = argparse.Namespace(url=None, version=False)
reply = cli.args_handler(args) reply = cli.args_handler(args)
assert ("waybackpy %s \nSee 'waybackpy --help" % (__version__)) == reply assert ("waybackpy %s" % (__version__)) in reply
def test_main(): def test_main():
# This also tests the parse_args method in cli.py # This also tests the parse_args method in cli.py