From 4862faa8f01585655202970889c442820b94242c Mon Sep 17 00:00:00 2001 From: moririn2528 Date: Wed, 16 Aug 2023 10:36:52 +0900 Subject: [PATCH] fix test --- python_tests/preferential/test_system_attrs.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/python_tests/preferential/test_system_attrs.py b/python_tests/preferential/test_system_attrs.py index b9e22e19..10448d48 100644 --- a/python_tests/preferential/test_system_attrs.py +++ b/python_tests/preferential/test_system_attrs.py @@ -17,12 +17,13 @@ def test_report_and_get_preferences(storage_supplier: Callable[[], StorageSuppli study.ask() study.ask() - assert len(get_preferences(study)) == 0 + study_id = study._study_id + assert len(get_preferences(study_id, storage)) == 0 better, worse = study.trials[0], study.trials[1] - report_preferences(study, [(better, worse)]) - assert len(get_preferences(study)) == 1 + report_preferences(study_id, storage, [(better.number, worse.number)]) + assert len(get_preferences(study_id, storage)) == 1 - actual_better, actual_worse = get_preferences(study)[0] - assert actual_better.number == better.number - assert actual_worse.number == worse.number + actual_better, actual_worse = get_preferences(study_id, storage)[0] + assert actual_better == better.number + assert actual_worse == worse.number