| 1365 | def test_custom_find_class(self): |
| 1366 | def loads(data): |
| 1367 | class Unpickler(self.unpickler): |
| 1368 | def find_class(self, module_name, global_name): |
| 1369 | return (module_name, global_name) |
| 1370 | return Unpickler(io.BytesIO(data)).load() |
| 1371 | |
| 1372 | self.assertEqual(loads(b'cmath\nlog\n.'), ('math', 'log')) |