test(countcomb): add tricky tests

serjzimmerman · Apr 17, 2024 · 0ae4c74 · 0ae4c74
1 parent 56d43ff
commit 0ae4c74
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/projects/chisel-practice/src/test/scala/projects/CountCombinations.scala b/projects/chisel-practice/src/test/scala/projects/CountCombinations.scala
@@ -6,10 +6,10 @@ import chiseltest._
 import org.scalatest.flatspec.AnyFlatSpec
 
 class CountCombinationsTester extends AnyFlatSpec with ChiselScalatestTester {
-  "CountCombinations of 0b11 in 0b11111111" should "be equal to 7" in {
+  "CountCombinations of 0b11 in 0b11111111" should "be equal to 4" in {
     test(new CountCombinations(8)) { dut =>
       dut.io.value.poke("b11111111".U)
-      dut.io.count.expect(7.U)
+      dut.io.count.expect(4.U)
     }
   }
 
@@ -19,6 +19,19 @@ class CountCombinationsTester extends AnyFlatSpec with ChiselScalatestTester {
       dut.io.count.expect(0.U)
     }
   }
+
+  "CountCombinations of 0b11 in 0b00000111" should "be equal to 1" in {
+    test(new CountCombinations(8)) { dut =>
+      dut.io.value.poke("b00000111".U)
+      dut.io.count.expect(1.U)
+    }
+  }
+  "CountCombinations of 0b11 in 0b11100111" should "be equal to 2" in {
+    test(new CountCombinations(8)) { dut =>
+      dut.io.value.poke("b11100111".U)
+      dut.io.count.expect(2.U)
+    }
+  }
 }
 
 class CountCombinationsTestbench(