From bef9efabc3f899a8f05cc6bee009c7c5fb3d01ae Mon Sep 17 00:00:00 2001 From: Nikita Sobolev Date: Mon, 7 Nov 2022 05:56:41 +0300 Subject: GH-99155: Fix `NormalDist` pickle with `0` and `1` protocols (GH99156) --- Lib/test/test_statistics.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'Lib/test/test_statistics.py') diff --git a/Lib/test/test_statistics.py b/Lib/test/test_statistics.py index 05ce79f1265..31a3cb6b53a 100644 --- a/Lib/test/test_statistics.py +++ b/Lib/test/test_statistics.py @@ -3003,14 +3003,19 @@ class TestNormalDist: nd = NormalDist(100, 15) self.assertNotEqual(nd, lnd) - def test_pickle_and_copy(self): + def test_copy(self): nd = self.module.NormalDist(37.5, 5.625) nd1 = copy.copy(nd) self.assertEqual(nd, nd1) nd2 = copy.deepcopy(nd) self.assertEqual(nd, nd2) - nd3 = pickle.loads(pickle.dumps(nd)) - self.assertEqual(nd, nd3) + + def test_pickle(self): + nd = self.module.NormalDist(37.5, 5.625) + for proto in range(pickle.HIGHEST_PROTOCOL + 1): + with self.subTest(proto=proto): + pickled = pickle.loads(pickle.dumps(nd, protocol=proto)) + self.assertEqual(nd, pickled) def test_hashability(self): ND = self.module.NormalDist -- cgit v1.2.3