def __init__(self, method_name="runTest"):
super().__init__(method_name)
self._current_base_path = Path(__file__).parent.parent.resolve()
init_config(str(self._current_base_path / "set_locations.ini"))
ds = TrainDataset(ds_path=config.get('DataPaths', 'run_data'))
logger.info(f"Found unique users: {len(ds.get_users())}")
self.train_dataset = ds.create_dataset()
def test_train_dataset_creation(self):
ds = TrainDataset(ds_path=config.get('DataPaths', 'run_data'))
logger.info(f"Found unique users: {len(ds.get_users())}")
gaze_data = ds.create_dataset()
self.assertTrue(len(gaze_data) > 0)
logger.info(f"Length of dataset is: {gaze_data.shape} with columns: {gaze_data.columns}")
logger.info(f"Size of dataset is {sys.getsizeof(gaze_data) / 1048576} Mb.")
logger.info(f"Unique stimulus types: {gaze_data['stimulus_type'].unique()}")