TST: fixed some unit tests

Author: dmgav

pyxrf/core/tests/test_map_processing.py

@@ -47,9 +47,10 @@ def test_dask_client_create(tmpdir):
 
     # Set the number of workers to some strange number (11 is unusual)
     #   (pass another kwarg in addition to default)
-    client = dask_client_create(n_workers=11)
+    n_workers_set = 11
+    client = dask_client_create(n_workers=n_workers_set)
     n_workers = len(client.scheduler_info()["workers"])
-    assert n_workers == 11, "The number of workers was set incorrectly"
+    assert n_workers == n_workers_set, "The number of workers was set incorrectly"
     client.close()
 
     assert not os.path.exists(dask_worker_space_path), "Temporary directory was created in the current directory"

pyxrf/core/tests/test_quant_analysis.py

@@ -347,9 +347,11 @@ def test_get_quant_fluor_data_dict():
         ), "Generated object contains emission lines that are different from expected"
 
         mass_sum = sum([_["density"] for _ in quant_fluor_data_dict["element_lines"].values()])
-        assert (
-            mass_sum == mass_sum_expected
-        ), "The total mass (density) of the components is different from expected"
+        npt.assert_almost_equal(
+            mass_sum, 
+            mass_sum_expected,
+            err_msg="The total mass (density) of the components is different from expected"
+        )
 
 
 def gen_xrf_map_dict(nx=10, ny=5, elines=["S_K", "Au_M", "Fe_K"]):

pyxrf/model/tests/test_hdf5_file_operations.py

@@ -23,7 +23,7 @@ def _prepare_raw_dataset(N=5, M=10, K=4096):
     pos_data = np.zeros(shape=[2, N, M])
     pos_data[0, :, :] = np.broadcast_to(np.linspace(1, 1 + (M - 1) * 0.1, M), shape=[N, M])
     pos_data[1, :, :] = np.broadcast_to(
-        np.reshape(np.linspace(5, 5 + (N - 1) * 0.2, N), newshape=[N, 1]), shape=[N, M]
+        np.reshape(np.linspace(5, 5 + (N - 1) * 0.2, N), shape=[N, 1]), shape=[N, M]
     )
 
     data = {