fix: Update Python writer unit test and snapshots for resourceType change

Author: ryukzak

test/api/write-generator/__snapshots__/python.test.ts.snap

@@ -7,7 +7,7 @@ exports[`Python Writer Generator generates Patient resource in inMemoryOnly mode
 
 from __future__ import annotations
 from pydantic import BaseModel, ConfigDict, Field, PositiveInt
-from typing import List as PyList, Literal
+from typing import Any, List as PyList, Literal
 
 from fhir_types.hl7_fhir_r4_core.base import (\\
     Address, Attachment, BackboneElement, CodeableConcept, ContactPoint, HumanName, Identifier, Period, Reference
@@ -65,6 +65,9 @@ class Patient(DomainResource):
     photo: PyList[Attachment] | None = Field(None, alias="photo", serialization_alias="photo")
     telecom: PyList[ContactPoint] | None = Field(None, alias="telecom", serialization_alias="telecom")
 
+    def model_post_init(self, __context: Any) -> None:
+        self.__pydantic_fields_set__.add("resource_type")
+
     def to_json(self, indent: int | None = None) -> str:
         return self.model_dump_json(exclude_unset=True, exclude_none=True, indent=indent)
 

test/api/write-generator/multi-package/__snapshots__/cda.test.ts.snap

@@ -76,7 +76,7 @@ exports[`CDA Python Generation should generate ClinicalDocument type (promoted l
 
 from __future__ import annotations
 from pydantic import BaseModel, ConfigDict, Field, PositiveInt
-from typing import List as PyList, Literal
+from typing import Any, List as PyList, Literal
 
 
 
@@ -121,6 +121,9 @@ class ClinicalDocument(ANY):
     type_id: II | None = Field(None, alias="typeId", serialization_alias="typeId")
     version_number: INT | None = Field(None, alias="versionNumber", serialization_alias="versionNumber")
 
+    def model_post_init(self, __context: Any) -> None:
+        self.__pydantic_fields_set__.add("resource_type")
+
     def to_json(self, indent: int | None = None) -> str:
         return self.model_dump_json(exclude_unset=True, exclude_none=True, indent=indent)
 

test/api/write-generator/multi-package/__snapshots__/local-package.test.ts.snap

@@ -192,7 +192,7 @@ exports[`Local Package Folder - Multi-Package Generation Python Generation shoul
 
 from __future__ import annotations
 from pydantic import BaseModel, ConfigDict, Field, PositiveInt
-from typing import List as PyList, Literal
+from typing import Any, List as PyList, Literal
 
 from fhir_types.hl7_fhir_r4_core.base import Coding, Identifier, Reference
 from fhir_types.hl7_fhir_r4_core.domain_resource import DomainResource
@@ -214,6 +214,9 @@ class ExampleNotebook(DomainResource):
     tag: PyList[Coding] | None = Field(None, alias="tag", serialization_alias="tag")
     title: str = Field(alias="title", serialization_alias="title")
 
+    def model_post_init(self, __context: Any) -> None:
+        self.__pydantic_fields_set__.add("resource_type")
+
     def to_json(self, indent: int | None = None) -> str:
         return self.model_dump_json(exclude_unset=True, exclude_none=True, indent=indent)
 
@@ -231,7 +234,7 @@ exports[`Local Package Folder - Multi-Package Generation Python Generation shoul
 
 from __future__ import annotations
 from pydantic import BaseModel, ConfigDict, Field, PositiveInt
-from typing import List as PyList, Literal
+from typing import Any, List as PyList, Literal
 
 from fhir_types.hl7_fhir_r4_core.base import Extension, Narrative
 from fhir_types.hl7_fhir_r4_core.resource import Resource
@@ -252,6 +255,9 @@ class DomainResource(Resource):
     modifier_extension: PyList[Extension] | None = Field(None, alias="modifierExtension", serialization_alias="modifierExtension")
     text: Narrative | None = Field(None, alias="text", serialization_alias="text")
 
+    def model_post_init(self, __context: Any) -> None:
+        self.__pydantic_fields_set__.add("resource_type")
+
     def to_json(self, indent: int | None = None) -> str:
         return self.model_dump_json(exclude_unset=True, exclude_none=True, indent=indent)
 
@@ -269,7 +275,7 @@ exports[`Local Package Folder - Multi-Package Generation Python Generation shoul
 
 from __future__ import annotations
 from pydantic import BaseModel, ConfigDict, Field, PositiveInt
-from typing import List as PyList, Literal
+from typing import Any, List as PyList, Literal
 
 from fhir_types.hl7_fhir_r4_core.base import (\\
     Address, Attachment, BackboneElement, CodeableConcept, ContactPoint, HumanName, Identifier, Period, Reference
@@ -327,6 +333,9 @@ class Patient(DomainResource):
     photo: PyList[Attachment] | None = Field(None, alias="photo", serialization_alias="photo")
     telecom: PyList[ContactPoint] | None = Field(None, alias="telecom", serialization_alias="telecom")
 
+    def model_post_init(self, __context: Any) -> None:
+        self.__pydantic_fields_set__.add("resource_type")
+
     def to_json(self, indent: int | None = None) -> str:
         return self.model_dump_json(exclude_unset=True, exclude_none=True, indent=indent)
 

test/api/write-generator/multi-package/__snapshots__/sql-on-fhir.test.ts.snap

@@ -86,7 +86,7 @@ exports[`SQL-on-FHIR Python Generation should generate ViewDefinition type (prom
 
 from __future__ import annotations
 from pydantic import BaseModel, ConfigDict, Field, PositiveInt
-from typing import List as PyList, Literal
+from typing import Any, List as PyList, Literal
 
 from fhir_types.hl7_fhir_r5_core.base import BackboneElement
 from fhir_types.hl7_fhir_r5_core.canonical_resource import CanonicalResource
@@ -162,6 +162,9 @@ class ViewDefinition(CanonicalResource):
     select: PyList[ViewDefinitionSelect] = Field(alias="select", serialization_alias="select")
     where: PyList[ViewDefinitionWhere] | None = Field(None, alias="where", serialization_alias="where")
 
+    def model_post_init(self, __context: Any) -> None:
+        self.__pydantic_fields_set__.add("resource_type")
+
     def to_json(self, indent: int | None = None) -> str:
         return self.model_dump_json(exclude_unset=True, exclude_none=True, indent=indent)
 
@@ -179,7 +182,7 @@ exports[`SQL-on-FHIR Python Generation should generate domain_resource for R5 1`
 
 from __future__ import annotations
 from pydantic import BaseModel, ConfigDict, Field, PositiveInt
-from typing import List as PyList, Literal
+from typing import Any, List as PyList, Literal
 
 from fhir_types.hl7_fhir_r5_core.base import Extension, Narrative
 from fhir_types.hl7_fhir_r5_core.resource import Resource
@@ -200,6 +203,9 @@ class DomainResource(Resource):
     modifier_extension: PyList[Extension] | None = Field(None, alias="modifierExtension", serialization_alias="modifierExtension")
     text: Narrative | None = Field(None, alias="text", serialization_alias="text")
 
+    def model_post_init(self, __context: Any) -> None:
+        self.__pydantic_fields_set__.add("resource_type")
+
     def to_json(self, indent: int | None = None) -> str:
         return self.model_dump_json(exclude_unset=True, exclude_none=True, indent=indent)
 

test/api/write-generator/python.test.ts

@@ -34,7 +34,7 @@ describe("Python Writer Generator", async () => {
     });
     it("generates base.py with TypeVar import and declaration", async () => {
         const basePy = result.filesGenerated["generated/hl7_fhir_r4_core/base.py"];
-        expect(basePy).toContain("from typing import Generic, List as PyList, Literal");
+        expect(basePy).toContain("from typing import Any, Generic, List as PyList, Literal");
         expect(basePy).toContain("from typing_extensions import TypeVar");
         expect(basePy).toContain("T = TypeVar('T', bound=str, default=str)");
     });