Add working statement reduction

1 files changed, 85 insertions, 0 deletions

diff --git a/test/Reduce.hs b/test/Reduce.hs
index 5afae18..37a2604 100644
--- a/test/Reduce.hs
+++ b/test/Reduce.hs
@@ -26,6 +26,7 @@ reduceUnitTests :: TestTree
 reduceUnitTests = testGroup "Reducer tests"
     [ moduleReducerTest
     , modItemReduceTest
+    , halveStatementsTest
     , activeWireTest
     ]
 
@@ -33,6 +34,7 @@ activeWireTest :: TestTree
 activeWireTest = testCase "Active wires" $ do
     findActiveWires verilog1 \\ ["x", "y", "z", "w"] @?= []
     findActiveWires verilog2 \\ ["x", "y", "z"] @?= []
+    findActiveWires verilog3 \\ ["x", "y", "clk", "r1", "r2"] @?= []
     where
         verilog1 = head $ getVerilog [verilog|
 module top(y, x);
@@ -54,6 +56,89 @@ module top(y, x);
   assign z = 0;
 endmodule
 |]
+        verilog3 = head $ getVerilog [verilog|
+module top(clk, y, x);
+  input clk;
+  input x;
+  output y;
+  reg r1;
+  reg r2;
+  reg r3;
+  always @(posedge clk) begin
+    r1 <= r3;
+    r2 <= r1;
+  end
+
+  always @(posedge clk) begin
+    r1 <= r2;
+  end
+  assign y = {r1, r2, r3};
+endmodule
+|]
+
+halveStatementsTest :: TestTree
+halveStatementsTest = testCase "Statements" $ do
+    GenVerilog <$> halveStatements srcInfo1 @?= golden1
+    where
+        srcInfo1 = SourceInfo "top" [verilog|
+module top(clk, y, x);
+  input clk;
+  input x;
+  output y;
+  reg r1;
+  reg r2;
+  reg r3;
+  always @(posedge clk) begin
+    r1 <= r3;
+    r2 <= r1;
+    r3 <= r2;
+  end
+
+  always @(posedge clk) begin
+    r1 <= r2;
+    r2 <= r3;
+    r3 <= r1;
+  end
+  assign y = {r1, r2, r3};
+endmodule
+|]
+        golden1 = GenVerilog <$> Dual (SourceInfo "top" [verilog|
+module top(clk, y, x);
+  input clk;
+  input x;
+  output y;
+  reg r1;
+  reg r2;
+  reg r3;
+  always @(posedge clk) begin
+    r1 <= 1'b0;
+  end
+
+  always @(posedge clk) begin
+    r1 <= 1'b0;
+  end
+  assign y = {r1, 1'b0, 1'b0};
+endmodule
+|]) (SourceInfo "top" [verilog|
+module top(clk, y, x);
+  input clk;
+  input x;
+  output y;
+  reg r1;
+  reg r2;
+  reg r3;
+  always @(posedge clk) begin
+    r2 <= 1'b0;
+    r3 <= r2;
+  end
+
+  always @(posedge clk) begin
+    r2 <= r3;
+    r3 <= 1'b0;
+  end
+  assign y = {1'b0, r2, r3};
+endmodule
+|])
 
 modItemReduceTest :: TestTree
 modItemReduceTest = testCase "Module items" $ do