Skip to content

Commit f0e23ac

Browse files
author
Hamish Downer
committed
Use CaseInsensitiveDict in RunReport (de)serialize tests
to ensure that we do handle that ok
1 parent f1555ba commit f0e23ac

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/test_utils.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from unittest.mock import patch
55

66
import pytest
7+
from requests.structures import CaseInsensitiveDict
78

89
from sortition_algorithms.errors import InfeasibleQuotasError, SelectionError, SelectionMultilineError
910
from sortition_algorithms.features import FeatureValueMinMax
@@ -509,21 +510,20 @@ def test_run_report_with_multiline_error_deserialisation(self):
509510

510511
def test_run_report_with_infeasible_quotas_error_serialisation(self):
511512
report = RunReport()
512-
report.add_error(
513-
InfeasibleQuotasError(
514-
features={"feat1": {"value1": FeatureValueMinMax(min=2, max=4)}}, output=["line1", "line2"]
515-
)
516-
)
513+
features = CaseInsensitiveDict()
514+
features["feat1"] = CaseInsensitiveDict()
515+
features["feat1"]["value1"] = FeatureValueMinMax(min=2, max=4)
516+
report.add_error(InfeasibleQuotasError(features=features, output=["line1", "line2"]))
517517
serialised_form = report.serialize()
518518
assert "_data" in serialised_form
519519
assert len(serialised_form["_data"]) == 1
520520

521521
def test_run_report_with_infeasible_quotas_error_deserialisation(self):
522522
report = RunReport()
523-
error = InfeasibleQuotasError(
524-
features={"feat1": {"value1": FeatureValueMinMax(min=2, max=4)}}, output=["line1", "line2"]
525-
)
526-
report.add_error(error)
523+
features = CaseInsensitiveDict()
524+
features["feat1"] = CaseInsensitiveDict()
525+
features["feat1"]["value1"] = FeatureValueMinMax(min=2, max=4)
526+
report.add_error(InfeasibleQuotasError(features=features, output=["line1", "line2"]))
527527
serialised_form = report.serialize()
528528
deserialised_report = RunReport.deserialize(serialised_form)
529529

0 commit comments

Comments
 (0)