fix summary st, and support test dataset sink mode

4 years ago · 5404826aea
parent 97a10bfa7c
commit 5404826aea
2 changed files with 110 additions and 83 deletions
--- a/tests/st/summary/test_summary.py
+++ b/tests/st/summary/test_summary.py
--- a/tests/summary_utils.py
+++ b/tests/summary_utils.py
@ -54,6 +54,8 @@ class SummaryReader:
        """Read next event."""
        file_handler = self._file_handler
        header = file_handler.read(_HEADER_SIZE)
+        if not header:
+            return None
        data_len = struct.unpack('Q', header)[0]
        # Ignore crc check.
        file_handler.read(_HEADER_CRC_SIZE)