use isclose for memory assertion

stevebachmeier · stevebachmeier · commit 9ec86f363111 · 2025-03-17T16:46:15.000-06:00
diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py
@@ -302,7 +302,7 @@ def test_set_up_dask_client_default() -> None:
             )
     else:
         available_memory = psutil.virtual_memory().total / (1024 ** 3)
-    assert sum(worker["memory_limit"] / 1024**3 for worker in workers.values()) == available_memory
+    assert np.isclose(sum(worker["memory_limit"] / 1024**3 for worker in workers.values()), available_memory, rtol=0.01)
 
 
 def test_set_up_dask_client_custom() -> None:

Original file line number	Diff line number	Diff line change
`@@ -302,7 +302,7 @@ def test_set_up_dask_client_default() -> None:`
`302`	`302`	`)`
`303`	`303`	`else:`
`304`	`304`	`available_memory = psutil.virtual_memory().total / (1024 ** 3)`
`305`		`- assert sum(worker["memory_limit"] / 1024**3 for worker in workers.values()) == available_memory`
	`305`	`+ assert np.isclose(sum(worker["memory_limit"] / 1024**3 for worker in workers.values()), available_memory, rtol=0.01)`
`306`	`306`
`307`	`307`
`308`	`308`	`def test_set_up_dask_client_custom() -> None:`