def test_read(self):
base_dir = os.path.dirname(__file__)
with open(f"{base_dir}/files/epts_meta.xml", "rb") as meta_data_fp:
meta_data = load_meta_data(meta_data_fp)
with open(f"{base_dir}/files/epts_raw.txt", "rb") as raw_data:
iterator = read_raw_data(raw_data, meta_data)
with performance_logging("load"):
assert list(iterator)
def test_skip_sensors(self):
base_dir = os.path.dirname(__file__)
with open(f'{base_dir}/files/epts_meta.xml', 'rb') as meta_data_fp, \
open(f'{base_dir}/files/epts_raw.txt', 'rb') as raw_data:
meta_data = load_meta_data(meta_data_fp)
records = read_raw_data(raw_data, meta_data, sensor_ids=["heartbeat"])
data_frame = DataFrame.from_records(records)
assert 'player_home_22_max_heartbeat' in data_frame.columns
assert 'player_home_22_x' not in data_frame.columns
def test_skip_sensors(self):
base_dir = os.path.dirname(__file__)
with open(f"{base_dir}/files/epts_meta.xml",
"rb") as metadata_fp, open(f"{base_dir}/files/epts_raw.txt",
"rb") as raw_data:
metadata = load_metadata(metadata_fp)
records = read_raw_data(raw_data,
metadata,
sensor_ids=["heartbeat"])
data_frame = DataFrame.from_records(records)
assert "player_1_max_heartbeat" in data_frame.columns
assert "player_1_x" not in data_frame.columns
