aboutsummaryrefslogtreecommitdiff
path: root/src/test/scala/firrtl/backends/experimental/smt/Btor2Spec.scala
blob: 21e8289ebf593a22c748a3f6a393b19e5b5a983a (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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
// SPDX-License-Identifier: Apache-2.0

package firrtl.backends.experimental.smt

import org.scalatest.flatspec.AnyFlatSpec

class Btor2Spec extends AnyFlatSpec {
  behavior.of("btor2 backend")

  it should "convert a hello world module" in {
    val src =
      """circuit m:
        |  module m:
        |    input clock: Clock
        |    input a: UInt<8>
        |    output b: UInt<16>
        |    b <= a
        |    assert(clock, eq(a, b), UInt(1), "") : a_eq_b
        |""".stripMargin

    val expected =
      """1 sort bitvec 8
        |2 input 1 a
        |3 sort bitvec 16
        |4 uext 3 2 8
        |5 output 4 ; b
        |6 sort bitvec 1
        |7 uext 3 2 8
        |8 neq 6 7 4
        |9 bad 8 ; a_eq_b
        |""".stripMargin

    assert(SMTBackendHelpers.toBotr2Str(src) == expected)
  }

  it should "include FileInfo in the output" in {
    val src =
      """circuit m: @[circuit 0:0]
        |  module m: @[module 0:0]
        |    input clock: Clock @[clock 0:0]
        |    input a: UInt<8> @[a 0:0]
        |    output b: UInt<16> @[b 0:0]
        |    b <= a @[b_a 0:0]
        |    assert(clock, eq(a, b), UInt(1), "") @[assert 0:0]
        |""".stripMargin

    val expected =
      """; @[module 0:0]
        |1 sort bitvec 8
        |2 input 1 a ; @[a 0:0]
        |3 sort bitvec 16
        |4 uext 3 2 8
        |5 output 4 ; b @[b 0:0], @[b_a 0:0]
        |6 sort bitvec 1
        |7 uext 3 2 8
        |8 neq 6 7 4
        |9 bad 8 ; assert_0 @[assert 0:0]
        |""".stripMargin

    assert(SMTBackendHelpers.toBotr2Str(src) == expected)
  }
}