From 49802e4429fea6299f1ffbb8382af30ecc53e6d7 Mon Sep 17 00:00:00 2001 From: "David R. Reich" <43832476+SiQube@users.noreply.github.com> Date: Thu, 24 Oct 2024 07:11:48 +0200 Subject: [PATCH] hotfix: check whether public dataset has gaze files (#872) --- .../public_dataset_processing_test.py | 23 ++++++++++--------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/tests/integration/public_dataset_processing_test.py b/tests/integration/public_dataset_processing_test.py index 85e06c88a..e1d79de3a 100644 --- a/tests/integration/public_dataset_processing_test.py +++ b/tests/integration/public_dataset_processing_test.py @@ -39,14 +39,15 @@ def test_public_dataset_processing(dataset_name, tmp_path): dataset.load() # Do some basic transformations. - if 'pixel' in dataset.gaze[0].columns: - dataset.pix2deg() - dataset.pos2vel() - dataset.pos2acc() - - for gaze in dataset.gaze: - assert 'position' in gaze.columns - assert 'velocity' in gaze.columns - assert 'acceleration' in gaze.columns - - shutil.rmtree(dataset_path, ignore_errors=True) + if dataset.definition.has_files['gaze']: + if 'pixel' in dataset.gaze[0].columns: + dataset.pix2deg() + dataset.pos2vel() + dataset.pos2acc() + + for gaze in dataset.gaze: + assert 'position' in gaze.columns + assert 'velocity' in gaze.columns + assert 'acceleration' in gaze.columns + + shutil.rmtree(dataset_path, ignore_errors=True)