Update query_test.py

- try and fix test
mattpitkin · Feb 16, 2024 · 0376646 · 0376646
1 parent e59bbdd
commit 0376646
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/psrqpy/tests/query_test.py b/psrqpy/tests/query_test.py
@@ -921,12 +921,12 @@ def test_derived_gw_parameters(query):
 
     # use query table functions
     ell = query.gw_ellipticity()
-    ellipticity = ell[ell["PSRJ"] == "J0835-4510"]["ELL"]
+    ellipticity = ell[ell["PSRJ"] == "J0835-4510"]["ELL"][0]
     assert 0.95 < ellipticity / expectedell < 1.05
 
     # use query table functions
     q22t = query.gw_mass_quadrupole()
-    q22 = q22t[q22t["PSRJ"] == "J0835-4510"]["Q22"]
+    q22 = q22t[q22t["PSRJ"] == "J0835-4510"]["Q22"][0]
     assert 0.95 < q22 / expectedQ22 < 1.05