Commit6162f8d

authored and

committed

Moved unit test to TestGenericPytree

1 parent257cc67 commit6162f8dCopy full SHA for 6162f8d

File tree

+15

-8

lines changed

+15

-8

lines changed

Lines changed: 15 additions & 8 deletions

Original file line number	Diff line number	Diff line change
`@@ -859,6 +859,21 @@ class DirectNamedTuple2(NamedTuple):`
`859`	`859`	`self.assertFalse(pytree.is_namedtuple(cls))`
`860`	`860`	`self.assertFalse(pytree.is_namedtuple_class(cls))`
`861`	`861`
	`862`	`+@parametrize(`
	`863`	`+"pytree",`
	`864`	`+ [`
	`865`	`+subtest(py_pytree,name="py"),`
	`866`	`+subtest(cxx_pytree,name="cxx"),`
	`867`	`+ ],`
	`868`	`+ )`
	`869`	`+deftest_enum_treespec_roundtrip(self,pytree):`
	`870`	`+data= {TestEnum.A:5}`
	`871`	`+spec=pytree.tree_structure(data)`
	`872`	`+`
	`873`	`+serialized=pytree.treespec_dumps(spec)`
	`874`	`+deserialized_spec=pytree.treespec_loads(serialized)`
	`875`	`+self.assertEqual(spec,deserialized_spec)`
	`876`	`+`
`862`	`877`
`863`	`878`	`classTestPythonPytree(TestCase):`
`864`	`879`	`deftest_deprecated_register_pytree_node(self):`
`@@ -1096,14 +1111,6 @@ def test_pytree_serialize_enum(self):`
`1096`	`1111`	`serialized_spec=py_pytree.treespec_dumps(spec)`
`1097`	`1112`	`self.assertIsInstance(serialized_spec,str)`
`1098`	`1113`
`1099`		`-deftest_enum_treespec_roundtrip(self):`
`1100`		`-data= {TestEnum.A:5}`
`1101`		`-spec=py_pytree.tree_structure(data)`
`1102`		`-`
`1103`		`-serialized=py_pytree.treespec_dumps(spec)`
`1104`		`-deserialized_spec=py_pytree.treespec_loads(serialized)`
`1105`		`-self.assertEqual(spec,deserialized_spec)`
`1106`		`-`
`1107`	`1114`	`deftest_pytree_serialize_namedtuple(self):`
`1108`	`1115`	`Point1=namedtuple("Point1", ["x","y"])`
`1109`	`1116`	`py_pytree._register_namedtuple(`

Comments

(0)