bt2: remove unused imports

[babeltrace.git] / tests / bindings / python / bt2 / test_message_iterator.py

diff --git a/tests/bindings/python/bt2/test_message_iterator.py b/tests/bindings/python/bt2/test_message_iterator.py
index 3cd27e5b77325536a7c8735756edd539bf76ec01..9635672fe792b409e53bf81cb8bc68036be81439 100644 (file)
--- a/tests/bindings/python/bt2/test_message_iterator.py
+++ b/tests/bindings/python/bt2/test_message_iterator.py
@@ -16,11 +16,9 @@
 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 #
 
-from bt2 import value
-import collections
 import unittest
-import copy
 import bt2
+import sys
 from utils import TestOutputPortMessageIterator
 
 
@@ -405,10 +403,12 @@ class UserMessageIteratorTestCase(unittest.TestCase):
         src = graph.add_component(MySource, 'src')
         it = TestOutputPortMessageIterator(graph, src.output_ports['out'])
 
-        # The initial refcount of Py_None was in the 7000, so 100000 iterations
-        # should be enough to catch the bug even if there are small differences
+        # Three times the initial ref count of `None` iterations should
+        # be enough to catch the bug even if there are small differences
         # between configurations.
-        for i in range(100000):
+        none_ref_count = sys.getrefcount(None) * 3
+
+        for i in range(none_ref_count):
             with self.assertRaises(bt2.TryAgain):
                 next(it)