assertion fix

SamoraHunter · SamoraHunter · commit 72f817d8d4c2 · 2025-03-12T06:35:22.000Z
diff --git a/notebooks/example_usage.ipynb b/notebooks/example_usage.ipynb
@@ -684,15 +684,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 55,
+   "execution_count": 29,
    "metadata": {},
    "outputs": [],
    "source": [
-    "\n",
-    "for col in dfmdi.select_dtypes(include=[np.number]).columns:\n",
-    "    assert np.allclose(dfmdi[col].values, merged_drugs[col].values, rtol=1e-5, atol=1e-8), f\"Mismatch in column: {col}\"\n",
-    "\n",
-    "assert dfmdi.select_dtypes(exclude=[np.number]).equals(merged_drugs.select_dtypes(exclude=[np.number]))\n"
+    "for col in dfmdi.select_dtypes(exclude=[np.number]).columns:\n",
+    "    assert dfmdi[col].astype(str).equals(merged_drugs[col].astype(str)), f\"Mismatch in column: {col}\""
    ]
   },
   {

Original file line number	Diff line number	Diff line change
`@@ -684,15 +684,12 @@`
`684`	`684`	`},`
`685`	`685`	`{`
`686`	`686`	`"cell_type": "code",`
`687`		`- "execution_count": 55,`
	`687`	`+ "execution_count": 29,`
`688`	`688`	`"metadata": {},`
`689`	`689`	`"outputs": [],`
`690`	`690`	`"source": [`
`691`		`- "\n",`
`692`		`- "for col in dfmdi.select_dtypes(include=[np.number]).columns:\n",`
`693`		`- " assert np.allclose(dfmdi[col].values, merged_drugs[col].values, rtol=1e-5, atol=1e-8), f\"Mismatch in column: {col}\"\n",`
`694`		`- "\n",`
`695`		`- "assert dfmdi.select_dtypes(exclude=[np.number]).equals(merged_drugs.select_dtypes(exclude=[np.number]))\n"`
	`691`	`+ "for col in dfmdi.select_dtypes(exclude=[np.number]).columns:\n",`
	`692`	`+ " assert dfmdi[col].astype(str).equals(merged_drugs[col].astype(str)), f\"Mismatch in column: {col}\""`
`696`	`693`	`]`
`697`	`694`	`},`
`698`	`695`	`{`