flake8

devito/ir/xdsl_iet/cluster_to_ssa.py

@@ -296,11 +296,11 @@ def _visit_math_nodes(self, dim: SteppingDimension, node: Expr,
 
         elif isinstance(node, cos):
             assert len(node.args) == 1, "Expected single argument for cos."
-            # import pdb; pdb.set_trace()
+
             return math.CosOp(self._visit_math_nodes(dim, node.args[0],
                               output_indexed)).result
 
-            # import pdb; pdb.set_trace()
+
             # return reduce(lambda x, y : arith.AndI(x, y).result, SSAargs)
 
         elif isinstance(node, Relational):
@@ -515,7 +515,7 @@ def build_time_loop(
     def lower_devito_Eqs(self, eqs: list[Any], **kwargs):
         # Lower devito Equations to xDSL
 
-        # import pdb; pdb.set_trace()
+
         for eq in eqs:
             lowered = self.operator._lower_exprs(as_tuple(eq), **kwargs)
             if isinstance(eq, Eq):
@@ -648,7 +648,7 @@ def convert(self, eqs: Iterable[Eq], **kwargs) -> ModuleOp:
                         functions.add(f.function)
 
                 elif isinstance(eq, Injection):
-                    # import pdb; pdb.set_trace()
+
                     functions.add(eq.field.function)
                     for f in retrieve_functions(eq.expr):
                         if isinstance(f, PointSource):

tests/test_xdsl_op_correctness.py

@@ -187,7 +187,7 @@ def test_extraction_from_lifted_ispace(rotate):
 
 @pytest.mark.parametrize('shape', [(101, 101), (201, 201)])
 @pytest.mark.parametrize('nt', [10, 20, ])
-def test_dt2_script(shape, nt):
+def test_dt2_sources_script(shape, nt):
 
     spacing = (10., 10.)  # spacing of 10 meters
     nbl = 1  # number of pad layers