minor improvment to performance

pymarshaler/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '0.2.2'
+__version__ = '0.2.3'
 __all__ = ['Marshal', 'utils', 'arg_delegates', 'errors']
 
 from pymarshaler.marshal import Marshal

pymarshaler/marshal.py

@@ -22,9 +22,9 @@ def register(self, cls, delegate: ArgBuilderDelegate):
 
     def get_for(self, cls):
         try:
-            for val in self.registered_delegates:
-                if cls == val or issubclass(cls, val):
-                    return self.registered_delegates[val]
+            for delegate_cls, delegate in self.registered_delegates.items():
+                if cls == delegate_cls or issubclass(cls, delegate_cls):
+                    return delegate
             return None
         except TypeError:
             return None

setup.py

@@ -8,7 +8,7 @@
 
 setuptools.setup(
     name="pymarshaler",
-    version="0.2.2",
+    version="0.2.3",
     author="Hernan Romer",
     author_email="[email protected]",
     description="Package to marshal and unmarshal python objects",

tests/test_marshaling.py

@@ -1,7 +1,6 @@
 import json
 import unittest
 
-from pymarshaler import marshal
 from pymarshaler.errors import MissingFieldsError, UnknownFieldError
 from pymarshaler.marshal import Marshal
 from tests.test_classes import *