fix

chanzuckerberg · Oct 22, 2023 · 4ef905a · 4ef905a
1 parent cc2986f
commit 4ef905a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/unit/backend/wmg/api/common/test_rollup.py b/tests/unit/backend/wmg/api/common/test_rollup.py
@@ -405,12 +405,18 @@ def test__rollup(self, _, input_cell_counts_df, expected_cell_counts_df, input_g
         expected_cell_counts_df.reset_index(inplace=True)
 
         assert_frame_equal(
-            rolled_up_cell_counts_df.reset_index(drop=True), expected_cell_counts_df.reset_index(drop=True)
+            rolled_up_cell_counts_df.reset_index(drop=True),
+            expected_cell_counts_df.reset_index(drop=True),
+            check_dtype=False,
         )
 
         # sort the rolled up gene expression dataframe so that the correct rows are compared with
         # the expected gene expression rows in the assert call
         sort_columns_for_rolled_gene_expr_df = list(cell_counts_df_index_list) + ["gene_ontology_term_id"]
         rolled_up_gene_expr_df.sort_values(sort_columns_for_rolled_gene_expr_df, inplace=True)
 
-        assert_frame_equal(rolled_up_gene_expr_df.reset_index(drop=True), expected_gene_expr_df.reset_index(drop=True))
+        assert_frame_equal(
+            rolled_up_gene_expr_df.reset_index(drop=True),
+            expected_gene_expr_df.reset_index(drop=True),
+            check_dtype=False,
+        )