Update test_association_task.py array unit tests

Update unit tests in test_association_task.py to use np.testing.assert_array_equal for testing array equality.

Author: bsmartradio

tests/test_association_task.py

@@ -68,14 +68,8 @@ def test_run(self):
         self.assertEqual(len(results.matchedDiaSources),
                          len(self.diaObjects) - 1)
         self.assertEqual(len(results.unAssocDiaSources), 1)
-        for test_obj_id, expected_obj_id in zip(
-                results.matchedDiaSources["diaObjectId"].to_numpy(),
-                [1, 2, 3, 4]):
-            self.assertEqual(test_obj_id, expected_obj_id)
-        for test_obj_id, expected_obj_id in zip(
-                results.unAssocDiaSources["diaObjectId"].to_numpy(),
-                [0]):
-            self.assertEqual(test_obj_id, expected_obj_id)
+        np.testing.assert_array_equal(results.matchedDiaSources["diaObjectId"].values, [1, 2, 3, 4])
+        np.testing.assert_array_equal(results.unAssocDiaSources["diaObjectId"].values, [0])
 
     def test_run_trailed_sources(self):
         """Test the full task by associating a set of diaSources to
@@ -119,6 +113,7 @@ def test_associate_sources(self):
                 assoc_result.diaSources["diaObjectId"].to_numpy(),
                 [0, 1, 2, 3, 4]):
             self.assertEqual(test_obj_id, expected_obj_id)
+        np.testing.assert_array_equal(assoc_result.diaSources["diaObjectId"].values, [0, 1, 2, 3, 4])
 
     def test_score_and_match(self):
         """Test association between a set of sources and an existing