def testSetDataOutput(self): """ Test the setDataOutput method with different arguments """ # Test 1: default output edPlugin = EDPlugin() xsDataStringTest = XSDataString("Test") edPlugin.setDataOutput(xsDataStringTest) EDAssert.equal(xsDataStringTest.marshal(), edPlugin.getDataOutput().marshal(), "Test 1: default output") # Test 2: named output edPlugin = EDPlugin() xsDataStringTest = XSDataString("Test") edPlugin.setDataOutput(xsDataStringTest, "testData") EDAssert.equal(xsDataStringTest.marshal(), edPlugin.getDataOutput("testData")[0].marshal(), "Test 2: named output") # Test 3: several inputs with the same name edPlugin = EDPlugin() xsDataStringTest1 = XSDataString("Test1") xsDataStringTest2 = XSDataString("Test2") edPlugin.setDataOutput(xsDataStringTest1, "testData") edPlugin.setDataOutput(xsDataStringTest2, "testData") pyListDataOutput = edPlugin.getDataOutput("testData") EDAssert.equal(xsDataStringTest1.marshal(), pyListDataOutput[0].marshal(), "Test 3: several inputs with the same name, 1") EDAssert.equal(xsDataStringTest2.marshal(), pyListDataOutput[1].marshal(), "Test 3: several inputs with the same name, 2") # Test 6: test of delDataOutput edPlugin = EDPlugin() xsDataStringTest = XSDataString("Test") edPlugin.setDataOutput(xsDataStringTest.marshal()) edPlugin.delDataOutput() EDAssert.equal(None, edPlugin.getDataOutput(), "Test 6: test of delDataOutput")