Update test_figure_factory.py

Author: LiamConnors

packages/python/plotly/plotly/tests/test_optional/test_figure_factory/test_figure_factory.py

@@ -4310,20 +4310,24 @@ def test_optional_arguments(self):
 
 
 class TestHexbinMapbox(NumpyTestUtilsMixin, TestCaseNoTemplate):
-    def assert_dict_almost_equal(self, dict1, dict2, decimal=7):
+    def compare_list_values(self, list1, list2, decimal=7):
+        assert len(list1) == len(list2), "Lists are not of the same length."
+        for i in range(len(list1)):
+            if isinstance(list1[i], list):
+                self.compare_list_values(list1[i], list2[i], decimal=decimal)
+            elif isinstance(list1[i], float):
+                np.testing.assert_almost_equal(list1[i], list2[i], decimal=decimal)
+            else:
+                assert (
+                    list1[i] == list2[i]
+                ), f"Values at index {i} are not equal: {list1[i]} != {list2[i]}"
+
+    def compare_dict_values(self, dict1, dict2, decimal=7):
         for k, v in dict1.items():
             if isinstance(v, dict):
-                self.assert_dict_almost_equal(v, dict2[k], decimal=decimal)
+                self.compare_dict_values(v, dict2[k], decimal=decimal)
             elif isinstance(v, list):
-                for i in range(len(v)):
-                    if isinstance(v[i], float):
-                        np.testing.assert_almost_equal(
-                            v[i], dict2[k][i], decimal=decimal
-                        )
-                    else:
-                        assert (
-                            v[i] == dict2[k][i]
-                        ), f"Values at index {i} for key {k} are not equal: {v[i]} != {dict2[k][i]}"
+                self.compare_list_values(v, dict2[k], decimal=decimal)
             elif isinstance(v, float):
                 np.testing.assert_almost_equal(v, dict2[k], decimal=decimal)
             else:
@@ -4437,7 +4441,7 @@ def test_aggregation(self):
 
         actual_agg = [2.0, 2.0, 1.0, 3.0, 9.0]
 
-        self.assert_dict_almost_equal(fig1.data[0].geojson, actual_geojson)
+        self.compare_dict_values(fig1.data[0].geojson, actual_geojson)
         assert np.array_equal(fig1.data[0].z, actual_agg)
 
         fig2 = ff.create_hexbin_mapbox(