Update on "[pytree] support PyStructSequence types for Python pytree"

XuehaiPan · XuehaiPan · commit e52b263c8b55 · 2023-11-08T19:32:46.000+08:00
Resolves #75982 - #75982 cc zou3519 [ghstack-poisoned]
diff --git a/test/test_pytree.py b/test/test_pytree.py
@@ -237,6 +237,7 @@ def run_test(ddct):
 
             unflattened = pytree_impl.tree_unflatten(values, treespec)
             self.assertEqual(unflattened, ddct)
+            self.assertEqual(unflattened.default_factory, ddct.default_factory)
             self.assertTrue(isinstance(unflattened, defaultdict))
 
         run_test(defaultdict(list, {}))
