move wrapping of stage into StageHandler

[ieee754fpu.git] / src / add / test_buf_pipe.py

diff --git a/src/add/test_buf_pipe.py b/src/add/test_buf_pipe.py
index 7155c90894bcb0f00a8d12e8cbc03cfb4af336c7..1408164a02647ccdfabc35db1cdca056367c6306 100644 (file)
--- a/src/add/test_buf_pipe.py
+++ b/src/add/test_buf_pipe.py
@@ -635,8 +635,8 @@ class ExampleStageDelayCls(StageCls, Elaboratable):
 class ExampleBufDelayedPipe(BufferedHandshake):
 
     def __init__(self):
-        stage = ExampleStageDelayCls(valid_trigger=2)
-        BufferedHandshake.__init__(self, stage, stage_ctl=True)
+        self.stage = ExampleStageDelayCls(valid_trigger=2)
+        BufferedHandshake.__init__(self, self.stage, stage_ctl=True)
 
     def elaborate(self, platform):
         m = BufferedHandshake.elaborate(self, platform)
@@ -667,8 +667,8 @@ def resultfn_12(data_o, expected, i, o):
 class ExampleUnBufDelayedPipe(BufferedHandshake):
 
     def __init__(self):
-        stage = ExampleStageDelayCls(valid_trigger=3)
-        BufferedHandshake.__init__(self, stage, stage_ctl=True)
+        self.stage = ExampleStageDelayCls(valid_trigger=3)
+        BufferedHandshake.__init__(self, self.stage, stage_ctl=True)
 
     def elaborate(self, platform):
         m = BufferedHandshake.elaborate(self, platform)