diff --git a/Lib/test/test_traceback.py b/Lib/test/test_traceback.py
index b7695d6..825a6d1 100644
--- a/Lib/test/test_traceback.py
+++ b/Lib/test/test_traceback.py
@@ -681,7 +681,10 @@ class TestStack(unittest.TestCase):
 
     def test_walk_stack(self):
         s = list(traceback.walk_stack(None))
-        self.assertGreater(len(s), 10)
+        if __name__ == "__main__":
+            self.assertEqual(len(s), 10)
+        else:
+            self.assertGreater(len(s), 10)
 
     def test_walk_tb(self):
         try: