#!/usr/bin/python2

execfile("onesquare.py")

from gnuradio import gr, gr_unittest
import numpy,sys

class qa_sq1(gr_unittest.TestCase):

    def test_001_gr_sq1(self):
        _src_data = [ 1,-0.1,5,-0.1,7,-0.1,-0.1,-0.1,1,1,-11,1 ]
        _expected_result = [ 1,-1,1,-1,1,-1,-1,-1,1,1,-1,1 ]

        src_data = []
        expected_result = []

        while len(src_data) < 90000:
            src_data += _src_data
            expected_result += _expected_result

        sq1 = onesquare(real_mode=True)

        src = gr.vector_source_f(src_data, False)
        dst = gr.vector_sink_f()

        tb = gr.top_block()
        tb.connect(src, sq1)
        tb.connect(sq1, dst)
        tb.run()

        result_data = dst.data()

        self.assertFloatTuplesAlmostEqual(expected_result, result_data, len(src_data))

if __name__ == '__main__':
    gr_unittest.main()