test: add df column type check

Author: ShriramS-Emerson

nisystemlink/clients/product/utilities/_dataframe_utilities.py

@@ -18,7 +18,9 @@ def convert_products_to_dataframe(products: List[Product]) -> DataFrame:
             - A new column would be created for unique properties across all products. The property
             columns would be named in the format `properties.property_name`.
     """
-    products_dict_representation = [product.dict(exclude_none=True) for product in products]
+    products_dict_representation = [
+        product.dict(exclude_none=True) for product in products
+    ]
     normalized_products_dataframe = pd.json_normalize(
         products_dict_representation, sep="."
     )

tests/product/test_product_dataframe_utilities.py

@@ -84,6 +84,11 @@ def test__convert_products_to_dataframe__with_complete_data(
             products_dataframe.columns.to_list()
             == expected_products_dataframe.columns.to_list()
         )
+        assert products_dataframe["updated_at"].dtype == "datetime64[ns]"
+        assert products_dataframe["file_ids"].dtype == "object"
+        assert isinstance(products_dataframe["file_ids"].iloc[0], List)
+        assert products_dataframe["keywords"].dtype == "object"
+        assert isinstance(products_dataframe["keywords"].iloc[0], List)
         pd.testing.assert_frame_equal(
             products_dataframe, expected_products_dataframe, check_dtype=True
         )