fix array comparison assertion issue in test

Becksteinlab · Aug 4, 2023 · 81a07ec · 81a07ec
1 parent 5118ada
commit 81a07ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mdpow/tests/test_workflows_base.py b/mdpow/tests/test_workflows_base.py
@@ -63,7 +63,7 @@ def test_guess_elements(self, create_universe):
         u = create_universe
         guessed_elements = base.guess_elements(u.atoms)
 
-        assert guessed_elements == self.check_guessed_elements
+        assert guessed_elements.all() == self.check_guessed_elements.all()
 
     def test_project_paths(self, test_df_data, project_paths_data):
         test_df = test_df_data