from gnuradio import gr, gr_unittest
from pylab import *

class qa_vector (gr_unittest.TestCase):

    def setUp (self):
        self.fg = gr.flow_graph ()

    def tearDown (self):
        self.fg = None

    def test_vector_ff (self):
        n = (-2, -1, 0, 1, 2)
        src_data = (-4, -2, 0, 2, 4)
	src = gr.vector_source_f(src_data)
	dst = gr.vector_sink_f()
	cst = gr.add_const_ff(2.0)
	self.fg.connect(src,cst)
	self.fg.connect(cst,dst)
	self.fg.run()
	dst_data = dst.data()
	print dst.data()
	
        markerline, stemlines, baseline = stem(n,dst_data,'-.')
	setp(markerline, 'markerfacecolor', 'b')
	setp(baseline, 'color','g', 'linewidth', 1)
	axis([-3,4,-5,7])
	show()

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