commit 56112895d69131ee4f34d4e3e9406614313df57f
author Georg Brandl <georg@python.org> Sun Jan 20 13:59:46 2008 +0000
committer Georg Brandl <georg@python.org> Sun Jan 20 13:59:46 2008 +0000
tree b57e176cd54df7984553c6337390ce1712fb2292
parent 92058d29334fcb4d01ae2ab7c165aba2ef6da2af

#1648: add sys.gettrace() and sys.getprofile().

Lib/test/test_profilehooks.py

diff --git a/Lib/test/test_profilehooks.py b/Lib/test/test_profilehooks.py
index 53f882a..2eedfd9 100644
--- a/Lib/test/test_profilehooks.py
+++ b/Lib/test/test_profilehooks.py
@@ -4,6 +4,22 @@
 
 from test import test_support
 
+class TestGetProfile(unittest.TestCase):
+    def setUp(self):
+        sys.setprofile(None)
+
+    def tearDown(self):
+        sys.setprofile(None)
+
+    def test_empty(self):
+        assert sys.getprofile() == None
+
+    def test_setget(self):
+        def fn(*args):
+            pass
+
+        sys.setprofile(fn)
+        assert sys.getprofile() == fn
 
 class HookWatcher:
     def __init__(self):
@@ -359,6 +375,7 @@
 
 def test_main():
     test_support.run_unittest(
+        TestGetProfile,
         ProfileHookTestCase,
         ProfileSimulatorTestCase
     )