log_read: return/yield all of input args from iteration

[pyi2ncommon] / test / test_log_read.py

diff --git a/test/test_log_read.py b/test/test_log_read.py
index 0fd4ae1..f5ebe61 100644 (file)
--- a/test/test_log_read.py
+++ b/test/test_log_read.py
@@ -200,9 +200,10 @@ class LogReadTester(unittest.TestCase):
                 reader = IterativeReader(file_handle)
                 self.helper_test_len(reader, 1)
                 counter = -1        # we may have to adapt this manually
-                for desc, text in reader:
+                for desc, text, source_idx in reader:
                     receive_time = perf_counter()
                     self.assertEqual(desc, self.temp_file)
+                    self.assertEqual(source_idx, 0)
                     counter += 1
                     text = text.strip()
                     if DEBUG:
@@ -255,7 +256,7 @@ class LogReadTester(unittest.TestCase):
             reader = LineReader(file_handle)
             self.helper_test_len(reader, 1)
 
-            for line_expected, (_, line_read) in zip(lines_expected, reader):
+            for line_expected, (_, line_read, _) in zip(lines_expected, reader):
                 if 'end' in line_read:
                     break
                 else: