def test_set_arguments(self): self.initialize_arguments() definition = InputDefinition() definition.set_arguments([self.foo]) self.assertEqual([self.foo], definition.get_arguments()) definition.set_arguments([self.bar]) self.assertEqual([self.bar], definition.get_arguments())