blob: 6365a1fa25adaa7ba4d3978f413a014cbed25411 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
// SPDX-License-Identifier: Apache-2.0
package firrtlTests.stage
import firrtl.stage.RunFirrtlTransformAnnotation
import firrtl.options.Shell
import firrtl.stage.FirrtlCli
import org.scalatest.flatspec.AnyFlatSpec
import org.scalatest.matchers.should.Matchers
class FirrtlCliSpec extends AnyFlatSpec with Matchers {
behavior.of("FirrtlCli for RunFirrtlTransformAnnotation / -fct / --custom-transforms")
it should "preserver transform order" in {
val shell = new Shell("foo") with FirrtlCli
val args = Array(
"--custom-transforms",
"firrtl.transforms.BlackBoxSourceHelper,firrtl.transforms.CheckCombLoops",
"--custom-transforms",
"firrtl.transforms.CombineCats",
"--custom-transforms",
"firrtl.transforms.ConstantPropagation"
)
val expected = Seq(
classOf[firrtl.transforms.BlackBoxSourceHelper],
classOf[firrtl.transforms.CheckCombLoops],
classOf[firrtl.transforms.CombineCats],
classOf[firrtl.transforms.ConstantPropagation]
)
shell
.parse(args)
.collect { case a: RunFirrtlTransformAnnotation => a }
.zip(expected)
.map { case (RunFirrtlTransformAnnotation(a), b) => a.getClass should be(b) }
}
}
|
