Ensure that the new pickle protocol support works with 'local names' (e.g. dotted method names).

Author: braxtonmckee

typed_python/SerializationContext.py

@@ -44,9 +44,16 @@ def pickledByStr(module_name: str, name: str) -> None:
     This mimics pickle's behavior when given a string from __reduce__. The
     string is interpreted as the name of a global variable, and pickle.whichmodules
     is used to search the module namespace, generating module_name.
+
+    Note that 'name' might contain '.' inside of it, since its a 'local name'.
     """
     module = importlib.import_module(module_name)
-    return getattr(module, name)
+
+    instance = module
+    for subName in name.split('.'):
+        instance = getattr(instance, subName)
+
+    return instance
 
 
 def createFunctionWithLocalsAndGlobals(code, globals):

typed_python/types_serialization_test.py

@@ -15,6 +15,8 @@
 import sys
 import os
 import importlib
+from functools import lru_cache
+
 from abc import ABC, abstractmethod, ABCMeta
 from typed_python.test_util import callFunctionInFreshProcess
 import typed_python.compiler.python_ast_util as python_ast_util
@@ -57,6 +59,13 @@
 module_level_testfun = dummy_test_module.testfunction
 
 
+class GlobalClassWithLruCache:
+    @staticmethod
+    @lru_cache(maxsize=None)
+    def f(x):
+        return x
+
+
 def moduleLevelFunctionUsedByExactlyOneSerializationTest():
     return "please don't touch me"
 
@@ -3085,3 +3094,10 @@ def f(self):
         x = callFunctionInFreshProcess(getX, (), showStdout=True)
 
         assert x == SerializationContext().deserialize(SerializationContext().serialize(x))
+
+    def test_serialize_class_static_lru_cache(self):
+        s = SerializationContext()
+
+        assert (
+            s.deserialize(s.serialize(GlobalClassWithLruCache.f)) is GlobalClassWithLruCache.f
+        )