fix convergence and potential test

Author: Niek Wielders

test_autogalaxy/profiles/mass/dark/test_cnfw.py

@@ -14,15 +14,13 @@ def test__deflections_yx_2d_from():
 def test_convergence_2d_from():
     cnfw = ag.mp.cNFW(centre=(0.0, 0.0), kappa_s=0.01591814312464436, scale_radius=0.36, core_radius=0.036)
 
-    convergence_2d = cnfw.convergence_2d_from(grid=ag.Grid2DIrregular([[1.0, 0.0]]))
-    convergence_r = np.sqrt(convergence_2d[0]**2 + convergence_2d[1]**2)
+    convergence = cnfw.convergence_2d_from(grid=ag.Grid2DIrregular([[1.0, 0.0]]))
 
-    assert convergence_r == pytest.approx(0.0, 1.0e-4)
+    assert convergence == pytest.approx(0.0, 1.0e-4)
 
-def potential_2d_from():
+def test_potential_2d_from():
     cnfw = ag.mp.cNFW(centre=(0.0, 0.0), kappa_s=0.01591814312464436, scale_radius=0.36, core_radius=0.036)
 
-    potential_2d = cnfw.potential_2d_from(grid=ag.Grid2DIrregular([[1.0, 0.0]]))
-    potential_r = np.sqrt(potential_2d[0] ** 2 + potential_2d[1] ** 2)
+    potential = cnfw.potential_2d_from(grid=ag.Grid2DIrregular([[1.0, 0.0]]))
 
-    assert potential_r == pytest.approx(0.0, 1.0e-4)
+    assert potential == pytest.approx(0.0, 1.0e-4)