diff options
Diffstat (limited to 'src/test/scala/firrtl/testutils/LeanTransformSpec.scala')
| -rw-r--r-- | src/test/scala/firrtl/testutils/LeanTransformSpec.scala | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/test/scala/firrtl/testutils/LeanTransformSpec.scala b/src/test/scala/firrtl/testutils/LeanTransformSpec.scala index d3510326..2d1cad8d 100644 --- a/src/test/scala/firrtl/testutils/LeanTransformSpec.scala +++ b/src/test/scala/firrtl/testutils/LeanTransformSpec.scala @@ -35,6 +35,10 @@ class LeanTransformSpec(protected val transforms: Seq[TransformDependency]) actual should be(expected) finalState } + protected def removeSkip(c: ir.Circuit): ir.Circuit = { + def onStmt(s: ir.Statement): ir.Statement = s.mapStmt(onStmt) + c.mapModule(m => m.mapStmt(onStmt)) + } } private object LeanTransformSpec { |