@@ -973,43 +973,61 @@ def test_cms_medicare_trustees_package_builds_part_b_premium_fact():
973973 assert values_by_record [record_id ].measure .unit == "usd"
974974
975975
976- def test_hhs_acf_liheap_package_builds_household_count_fact ():
977- package = load_source_package ("hhs-acf-liheap-fy2024-national-profile" )
978- cells = package .build_source_cells (2024 )
979- records = package .build_source_records (2024 , cells = cells )
980- facts = package .build_facts (2024 , cells = cells )
976+ @pytest .mark .parametrize (
977+ ("source" , "year" , "cell_count" , "households" , "addresses" , "source_file" ),
978+ [
979+ (
980+ "hhs-acf-liheap-fy2023-national-profile" ,
981+ 2023 ,
982+ 360 ,
983+ 5_939_605 ,
984+ ("E5" , "E1" , "A5" , "B5" , "D5" , "F5" ),
985+ "acf_liheap_fy2023_all_states_national_profile.pdf" ,
986+ ),
987+ (
988+ "hhs-acf-liheap-fy2024-national-profile" ,
989+ 2024 ,
990+ 348 ,
991+ 5_876_646 ,
992+ ("E3" , "E1" , "A3" , "B3" , "D3" , "F3" ),
993+ "acf_liheap_fy2024_all_states_national_profile.pdf" ,
994+ ),
995+ ],
996+ )
997+ def test_hhs_acf_liheap_package_builds_household_count_fact (
998+ source ,
999+ year ,
1000+ cell_count ,
1001+ households ,
1002+ addresses ,
1003+ source_file ,
1004+ ):
1005+ package = load_source_package (source )
1006+ cells = package .build_source_cells (year )
1007+ records = package .build_source_records (year , cells = cells )
1008+ facts = package .build_facts (year , cells = cells )
9811009 values_by_record = {fact .source_record_id : fact for fact in facts }
9821010
983- assert package .package_id == "hhs-acf-liheap-fy2024-national-profile"
1011+ assert package .package_id == source
9841012 assert validate_source_cells (cells ).valid
9851013 assert validate_facts (facts ).valid
986- assert len (cells ) == 348
1014+ assert len (cells ) == cell_count
9871015 assert len (facts ) == 1
9881016 assert all (fact .source .raw_r2_uri for fact in facts )
9891017
9901018 record_id = (
991- "hhs_acf_liheap.fy2024 .national_profile."
1019+ f "hhs_acf_liheap.fy { year } .national_profile."
9921020 "state_programs.households_served"
9931021 )
994- assert records [0 ].source_cell_addresses == (
995- "E3" ,
996- "E1" ,
997- "A3" ,
998- "B3" ,
999- "D3" ,
1000- "F3" ,
1001- )
1002- assert values_by_record [record_id ].value == 5_876_646
1022+ assert records [0 ].source_cell_addresses == addresses
1023+ assert values_by_record [record_id ].value == households
10031024 assert values_by_record [record_id ].measure .concept == (
10041025 "hhs_acf_liheap.households_served_by_state_programs"
10051026 )
1006- assert values_by_record [record_id ].period .value == 2024
1027+ assert values_by_record [record_id ].period .value == year
10071028 assert values_by_record [record_id ].geography .id == "0100000US"
10081029 assert values_by_record [record_id ].entity .name == "household"
1009- assert (
1010- values_by_record [record_id ].source .source_file
1011- == "acf_liheap_fy2024_all_states_national_profile.pdf"
1012- )
1030+ assert values_by_record [record_id ].source .source_file == source_file
10131031 assert values_by_record [record_id ].constraints
10141032
10151033
0 commit comments