Add explicit tests for ohlcv regex
This commit is contained in:
@@ -30,8 +30,7 @@ class HDF5DataHandler(IDataHandler):
|
||||
"""
|
||||
_tmp = [
|
||||
re.search(
|
||||
r'^([a-zA-Z_]+(\:[a-zA-Z]{2,}(\-[0-9]{2,})?)?)\-(\d+\S)\-?([a-zA-Z_]*)?(?=.h5)',
|
||||
p.name
|
||||
cls._OHLCV_REGEX, p.name
|
||||
) for p in datadir.glob("*.h5")
|
||||
]
|
||||
return [(match[1].replace('_', '/'), match[2], match[3]) for match in _tmp
|
||||
|
@@ -23,6 +23,8 @@ logger = logging.getLogger(__name__)
|
||||
|
||||
class IDataHandler(ABC):
|
||||
|
||||
_OHLCV_REGEX = r'^([a-zA-Z_]+)\-(\d+\S+)(?=\.)'
|
||||
|
||||
def __init__(self, datadir: Path) -> None:
|
||||
self._datadir = datadir
|
||||
|
||||
|
@@ -31,8 +31,7 @@ class JsonDataHandler(IDataHandler):
|
||||
"""
|
||||
_tmp = [
|
||||
re.search(
|
||||
r'^([a-zA-Z_]+(\:[a-zA-Z]{2,}(\-[0-9]{2,})?)?)\-(\d+\S)\-?([a-zA-Z_]*)?(?=.json)',
|
||||
p.name
|
||||
cls._OHLCV_REGEX, p.name
|
||||
) for p in datadir.glob(f"*.{cls._get_file_extension()}")]
|
||||
return [(match[1].replace('_', '/'), match[2], match[3]) for match in _tmp
|
||||
if match and len(match.groups()) > 1]
|
||||
|
Reference in New Issue
Block a user