test: updated tests

alercebroker · Oct 20, 2023 · 62fbabe · 62fbabe
1 parent c81f2b2
commit 62fbabe
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/prv_candidates_step/prv_candidates_step/core/strategy/ztf.py b/prv_candidates_step/prv_candidates_step/core/strategy/ztf.py
@@ -23,8 +23,8 @@ def prv_forced_photometry_mapper() -> dict:
 
     mapping.update(
         {
-            "e_ra": Mapper(lambda x: 0),
-            "e_dec": Mapper(lambda x: 0),
+            "e_ra": Mapper(lambda: 0),
+            "e_dec": Mapper(lambda: 0),
             "isdiffpos": Mapper(
                 lambda x: 1 if x >= 0 else -1, origin="forcediffimflux"
             ),

diff --git a/sorting_hat_step/tests/unittest/test_wizard.py b/sorting_hat_step/tests/unittest/test_wizard.py
@@ -150,10 +150,10 @@ def test_decode(self):
     def test_write_object(self, insert_mock):
         alerts = pd.DataFrame(
             [
-                {"oid": 10, "aid": 0, "sid": "ZTF", "extra": "extra1"},
-                {"oid": 20, "aid": 1, "sid": "ZTF", "extra": "extra2"},
-                {"oid": 30, "aid": 0, "sid": "ZTF", "extra": "extra3"},
-                {"oid": 40, "aid": 2, "sid": "ZTF", "extra": "extra4"},
+                {"oid": 10, "aid": 0, "sid": "ZTF", "extra_fields": {}},
+                {"oid": 20, "aid": 1, "sid": "ZTF", "extra_fields": {}},
+                {"oid": 30, "aid": 0, "sid": "ZTF", "extra_fields": {}},
+                {"oid": 40, "aid": 2, "sid": "ZTF", "extra_fields": {}},
             ]
         )