corrected t7 tests

raft-tech · Jan 10, 2024 · 620bd22 · 620bd22
1 parent e45cc62
commit 620bd22
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tdrs-backend/tdpservice/parsers/schema_defs/tribal_tanf/t7.py b/tdrs-backend/tdpservice/parsers/schema_defs/tribal_tanf/t7.py
@@ -1,8 +1,7 @@
 """Schema for Tribal TANF T7 Row."""
 
-from tdpservice.parsers.util import SchemaManager
 from tdpservice.parsers.fields import Field, TransformField
-from tdpservice.parsers.row_schema import RowSchema
+from tdpservice.parsers.row_schema import RowSchema, SchemaManager
 from tdpservice.parsers.transforms import calendar_quarter_to_rpt_month_year
 from tdpservice.parsers import validators
 from tdpservice.search_indexes.models.tribal import Tribal_TANF_T7

diff --git a/tdrs-backend/tdpservice/parsers/test/test_parse.py b/tdrs-backend/tdpservice/parsers/test/test_parse.py
@@ -1203,6 +1203,8 @@ def tribal_section_4_file(stt_user, stt):
 @pytest.mark.django_db()
 def test_parse_tribal_section_4_file(tribal_section_4_file):
     """Test parsing Tribal TANF Section 4 submission."""
+    tribal_section_4_file.year = 2020
+    tribal_section_4_file.quarter = 'Q1'
     parse.parse_datafile(tribal_section_4_file)
 
     assert Tribal_TANF_T7.objects.all().count() == 18