fix pytest, v0.1.1

Author: AvivAbramovich

Makefile

@@ -9,5 +9,7 @@ clean:
 	rm -rf dist build argreq.egg-info
 upload:
 	twine upload dist/*
+test:
+	pytest
 rst:
 	m2r --overwrite README.md

argreq/__init__.py

@@ -6,7 +6,7 @@
 
 import argreq.errors
 
-__VERSION__ = '0.1.0'
+__VERSION__ = '0.1.1'
 
 
 def _find_argument(
@@ -35,7 +35,7 @@ def check(_req):
         else:
             req_with_arg = f'value{_req}'
         if not eval(req_with_arg):
-            raise NotValidArgumentError(value, _req)
+            raise argreq.errors.NotValidArgumentError(value, _req)
     if builtins.type(req) is tuple:
         for _req in req:
             check(_req)

test_example.py

@@ -1,9 +1,10 @@
-from argreq import argument, NotValidArgumentError
+from argreq import argument
+from argreq.errors import NotValidArgumentError
 import functools
 import pytest
 
 def return_true(func):
-    functools.wraps(func)
+    @functools.wraps(func)
     def wrapper(*args, **kwargs):
         res = func(*args, **kwargs)
         if res is None: