Allow passing reference epoch to register reader

Author: glopesdev

harp/reader.py

@@ -3,6 +3,7 @@
 from math import log2
 from os import PathLike
 from pathlib import Path
+from datetime import datetime
 from functools import partial
 from numpy import dtype
 from pandas import DataFrame, Series
@@ -17,7 +18,9 @@
 
 class _ReadRegister(Protocol):
     def __call__(
-        self, file: Optional[Union[str, bytes, PathLike[Any], BinaryIO]] = None
+        self,
+        file: Optional[Union[str, bytes, PathLike[Any], BinaryIO]] = None,
+        epoch: Optional[datetime] = None,
     ) -> DataFrame:
         ...
 
@@ -134,6 +137,7 @@ def _create_register_reader(register: Register, base_path: Path):
     def reader(
         file: Optional[Union[str, bytes, PathLike[Any], BinaryIO]] = None,
         columns: Optional[Axes] = None,
+        epoch: Optional[datetime] = None,
     ):
         if file is None:
             file = f"{base_path}_{register.address}.bin"
@@ -144,6 +148,7 @@ def reader(
             dtype=dtype(register.type),
             length=register.length,
             columns=columns,
+            epoch=epoch,
         )
         return data
 

tests/test_reader.py

@@ -12,7 +12,7 @@
         path="data/device.yml",
         expected_whoAmI=0,
         expected_registers=["DigitalInputMode"],
-    )
+    ),
 ]