def test_003_square3_ff(self):
src_data0 = (0, 1, 0, 1, 0)
src_data1 = (-3.0, 4.0, -5.5, 2.0, 3.0)
src_data2 = (2.0, 2.0, 2.0, 2.0, 2.0)
src_data3 = (2.7, 2.7, 2.1, 2.3, 2.5)
expected_result0 = (-3.0, 2.0, -5.5, 2.0, 3.0)
expected_result1 = (-3.0, 2.7, -5.5, 2.3, 3.0)
src0 = blocks.vector_source_f(src_data0)
src1 = blocks.vector_source_f(src_data1)
src2 = blocks.vector_source_f(src_data2)
src3 = blocks.vector_source_f(src_data3)
sqr = square3_ff()
dst0 = blocks.vector_sink_f()
dst1 = blocks.vector_sink_f()
self.tb.connect(src0, (sqr, 0))
self.tb.connect(src1, (sqr, 1))
self.tb.connect(src2, (sqr, 2))
self.tb.connect(src3, (sqr, 3))
self.tb.connect((sqr, 0), dst0)
self.tb.connect((sqr, 1), dst1)
self.tb.run()
result_data0 = dst0.data()
result_data1 = dst1.data()
from pudb import set_trace
set_trace()
self.assertFloatTuplesAlmostEqual(expected_result0, result_data0, 6)
self.assertFloatTuplesAlmostEqual(expected_result1, result_data1, 6)
def test_003_square3_ff(self):
src_data = (-3, 4, -5.5, 2, 3)
expected_result = (9, 16, 30.25, 4, 9)
src = blocks.vector_source_f(src_data)
sqr = square3_ff()
dst = blocks.vector_sink_f()
self.tb.connect(src, sqr)
self.tb.connect(sqr, dst)
self.tb.run()
result_data = dst.data()
self.assertFloatTuplesAlmostEqual(expected_result, result_data, 6)
def test_003_square3_ff (self):
src_data = (-3, 4, -5.5, 2, 3)
expected_result = (9, 16, 30.25, 4, 9)
src = gr.vector_source_f (src_data)
sqr = square3_ff ()
dst = gr.vector_sink_f ()
self.tb.connect (src, sqr)
self.tb.connect (sqr, dst)
self.tb.run ()
result_data = dst.data ()
self.assertFloatTuplesAlmostEqual (expected_result, result_data, 6)
def test_001_t (self):
src_data = (0,0,0,0,0,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1)
expected_result = (-2.0, 0.0, 5.0, 8.0, 9.0, 11.0, 14.0, 18.0)
src0 = gr.vector_source_f(src_data)
sqr = square3_ff()
#Preload
sqr.input_config(1).preload_items = 1
dst = gr.vector_sink_f()
self.tb.connect(src0, (sqr,0)) # src0(vector_source) -> sqr_input_0
#self.tb.connect((sqr,0), (sqr,1)) # sqr_output_0 -> sqr_input_1
self.tb.connect(sqr,dst) # sqr_output_0 -> dst (vector_source)
self.tb.run()
result_data = dst.data()
print str(result_data), "Result data"
print str(expected_result), "expected "
def test_001_t(self):
src_data = (0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1, 1, 1, 1)
expected_result = (-2.0, 0.0, 5.0, 8.0, 9.0, 11.0, 14.0, 18.0)
src0 = gr.vector_source_f(src_data)
sqr = square3_ff()
#Preload
sqr.input_config(1).preload_items = 1
dst = gr.vector_sink_f()
self.tb.connect(src0, (sqr, 0)) # src0(vector_source) -> sqr_input_0
#self.tb.connect((sqr,0), (sqr,1)) # sqr_output_0 -> sqr_input_1
self.tb.connect(sqr, dst) # sqr_output_0 -> dst (vector_source)
self.tb.run()
result_data = dst.data()
print str(result_data), "Result data"
print str(expected_result), "expected "
