Merge pull request #19 from matthewwardrop/fix_nested_type_resolution

Fix nested type resolution in spec-classes.

Author: matthewwardrop

spec_classes/spec_class.py

@@ -261,7 +261,7 @@ def bootstrap(self, spec_cls: type):
         managed_attrs.extend(self.attrs)
 
         # Generate namespace of annotations (in addition to local class context)
-        annotation_namespace = {spec_cls.__name__: spec_cls}
+        annotation_namespace = {spec_cls.__name__: spec_cls, **vars(spec_cls)}
         if hasattr(spec_cls, "ANNOTATION_TYPES"):
             spec_annotation_types = spec_cls.ANNOTATION_TYPES
             if hasattr(spec_annotation_types, "__call__"):

tests/test_spec_class.py

@@ -879,3 +879,13 @@ class SubSpec(Spec):
             a = staticmethod(lambda x: x)
 
         SubSpec()  # Would fail if attempting to deepcopy `staticmethod`.
+
+    def test_nested_type_resolution(self):
+        @spec_class
+        class A:
+            class B:
+                pass
+
+            a: B
+
+        assert A.__spec_class__.attrs["a"].type is A.B