def test_should_add_new_datastream(self): tester = ComponentTester() tester.addDataStream('First', 'out_data', 'Second', 'in_data') with open(self.defaultFileName) as file: contents = file.read() assert_that(contents, contains_string('<Source name="First.out_data">\n\t\t\t<sink>Second.in_data</sink>'))
def test_should_check_component_output(self): tester = ComponentTester() # print('adding generator...') tester.addGenerator('SampleGenerators:CvMatGenerator', 'Generator1') tester.addGenerator('SampleGenerators:CvMatGenerator', 'Generator2') # print('adding component...') tester.setComponent('Summator', 'CvBasic:Sum') # print('adding component...') tester.addSink('SampleGenerators:CvMatSink') tester.addDataStream('Generator1', 'out_img', 'Summator', 'in_img1') tester.addDataStream('Generator2', 'out_img', 'Summator', 'in_img2') tester.addDataStream('Summator', 'out_img', 'Sink', 'in_img') tester.addTerminationStatement('END OF SEQUENCE') # print('Task body:') # print(tester.taskBuilder.getTaskBody()) tester.start() output = tester.getOutput() # print(output) # print('finished printing output') assert_that(output, contains_string('[2, 2, 2, 2;\n 2, 2, 2, 2;\n 2, 2, 2, 2]'))