ch25: sentinel metaclass example

ramalho · ramalho · commit 493b0d2a565b · 2021-05-23T22:30:40.000-03:00
diff --git a/25-class-metaprog/sentinel/sentinel_test.py b/25-class-metaprog/sentinel/sentinel_test.py
@@ -1,3 +1,5 @@
+import pickle
+
 from sentinel import Sentinel
 
 class PlainSentinel(Sentinel): pass
@@ -12,12 +14,17 @@ def test_repr():
 
 
 def test_pickle():
-    from pickle import dumps, loads
-    s = dumps(PlainSentinel)
-    ps = loads(s)
+    s = pickle.dumps(PlainSentinel)
+    ps = pickle.loads(s)
     assert ps is PlainSentinel
 
 
 def test_custom_repr():
     assert repr(SentinelCustomRepr) == '***SentinelRepr***'
-    
+
+
+def test_sentinel_comes_ready_to_use():
+    assert repr(Sentinel) == 'Sentinel'
+    s = pickle.dumps(Sentinel)
+    ps = pickle.loads(s)
+    assert ps is Sentinel
