set difference, maybe forgot to push?

This commit is contained in:
lucasmbrown-usds 2022-09-15 15:33:28 -04:00
parent 5f1fc28a8d
commit d5a5f3724a

View file

@ -364,7 +364,7 @@ def compare_to_list_of_expected_state_fips_codes(
# If continental US is not expected to be included, remove it from the # If continental US is not expected to be included, remove it from the
# expected states set. # expected states set.
if not continental_us_expected: if not continental_us_expected:
expected_states_set = expected_states_set - set( expected_states_set = expected_states_set.difference(
TILES_CONTINENTAL_US_FIPS_CODE TILES_CONTINENTAL_US_FIPS_CODE
) )
@ -374,27 +374,27 @@ def compare_to_list_of_expected_state_fips_codes(
# do not use this argument -- instead, use `additional_fips_codes_not_expected` # do not use this argument -- instead, use `additional_fips_codes_not_expected`
# for the 1 state you expected to be missing. # for the 1 state you expected to be missing.
if not alaska_and_hawaii_expected: if not alaska_and_hawaii_expected:
expected_states_set = expected_states_set - set( expected_states_set = expected_states_set.difference(
TILES_ALASKA_AND_HAWAII_FIPS_CODE TILES_ALASKA_AND_HAWAII_FIPS_CODE
) )
# If Puerto Rico is not expected to be included, remove it from the expected # If Puerto Rico is not expected to be included, remove it from the expected
# states set. # states set.
if not puerto_rico_expected: if not puerto_rico_expected:
expected_states_set = expected_states_set - set( expected_states_set = expected_states_set.difference(
TILES_PUERTO_RICO_FIPS_CODE TILES_PUERTO_RICO_FIPS_CODE
) )
# If island areas are not expected to be included, remove them from the expected # If island areas are not expected to be included, remove them from the expected
# states set. # states set.
if not island_areas_expected: if not island_areas_expected:
expected_states_set = expected_states_set - set( expected_states_set = expected_states_set.difference(
TILES_ISLAND_AREA_FIPS_CODES TILES_ISLAND_AREA_FIPS_CODES
) )
# If additional FIPS codes are not expected to be included, remove them from the # If additional FIPS codes are not expected to be included, remove them from the
# expected states set. # expected states set.
expected_states_set = expected_states_set - set( expected_states_set = expected_states_set.difference(
additional_fips_codes_not_expected additional_fips_codes_not_expected
) )