fix: use == to compare string literal

PyLops · Nov 17, 2024 · 571f6d5 · 571f6d5
1 parent d68ee79
commit 571f6d5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytests/test_oneway.py b/pytests/test_oneway.py
@@ -96,7 +96,7 @@ def test_PhaseShift_3dsignal(par):
 @pytest.mark.parametrize("par", [(par1), (par2), (par1v), (par2v)])
 def test_Deghosting_2dsignal(par, create_data2D):
     """Deghosting of 2d data"""
-    p2d, p2d_minus = create_data2D(1 if par["kind"] is "p" else -1)
+    p2d, p2d_minus = create_data2D(1 if par["kind"] == "p" else -1)
 
     p2d_minus_inv, p2d_plus_inv = Deghosting(
         p2d,
@@ -111,7 +111,7 @@ def test_Deghosting_2dsignal(par, create_data2D):
         npad=0,
         ntaper=0,
         dtype=par["dtype"],
-        **dict(damp=1e-10, iter_lim=60)
+        **dict(damp=1e-10, iter_lim=60),
     )
 
     assert np.linalg.norm(p2d_minus_inv - p2d_minus) / np.linalg.norm(p2d_minus) < 3e-1