fix tests that depend on Driver (#2429)

Co-authored-by: sinofp <sinofp@tuta.io>

1 files changed, 7 insertions, 4 deletions

diff --git a/src/test/scala/firrtlTests/CustomTransformSpec.scala b/src/test/scala/firrtlTests/CustomTransformSpec.scala
index 50cb6f43..30daa878 100644
--- a/src/test/scala/firrtlTests/CustomTransformSpec.scala
+++ b/src/test/scala/firrtlTests/CustomTransformSpec.scala
@@ -129,11 +129,14 @@ class CustomTransformSpec extends FirrtlFlatSpec {
   }
 
   they should "not cause \"Internal Errors\"" in {
-    val optionsManager = new ExecutionOptionsManager("test") with HasFirrtlOptions {
-      firrtlOptions = FirrtlExecutionOptions(firrtlSource = Some(input), customTransforms = List(new ErroringTransform))
-    }
     (the[java.lang.IllegalArgumentException] thrownBy {
-      Driver.execute(optionsManager)
+      (new FirrtlStage).execute(
+        Array(),
+        Seq(
+          FirrtlSourceAnnotation(input),
+          RunFirrtlTransformAnnotation(new ErroringTransform)
+        )
+      )
     }).getMessage should include(errorString)
   }