fix tests broken by taking lint advice

Author: zenmoto

tests/searchcommands/chunked_data_stream.py

@@ -42,11 +42,11 @@ def __init__(self, stream):
     def read_chunk(self):
         header = self.stream.readline()
 
-        while header > 0 and header.strip() == b'':
+        while len(header) > 0 and header.strip() == b'':
             header = self.stream.readline()  # Skip empty lines
-
-        if not header == 0:
+        if len(header) == 0:
             raise EOFError
+
         version, meta, data = header.rstrip().split(b',')
         metabytes = self.stream.read(int(meta))
         databytes = self.stream.read(int(data))

tests/searchcommands/test_generator_command.py

@@ -34,6 +34,9 @@ def generate(self):
         finished_seen = chunk.meta.get("finished", False)
         for row in chunk.data:
             seen.add(row["event_index"])
+    print(out_stream.getvalue())
+    print(expected)
+    print(seen)
     assert expected.issubset(seen)
     assert finished_seen