def test_get_invalid_argument(self):
self.initialize_arguments()
definition = InputDefinition()
definition.add_arguments([self.foo])
self.assertRaises(Exception, 'The "bar" argument does not exist.',
definition.get_argument, 'bar')
def test_add_arguments(self):
self.initialize_arguments()
definition = InputDefinition()
definition.add_arguments([self.foo])
self.assertEqual([self.foo], definition.get_arguments())
definition.add_arguments([self.bar])
self.assertEqual([self.foo, self.bar], definition.get_arguments())
def test_has_argument(self):
self.initialize_arguments()
definition = InputDefinition()
definition.add_arguments([self.foo])
self.assertTrue(definition.has_argument("foo"))
self.assertFalse(definition.has_argument("bar"))
def test_add_arguments(self):
self.initialize_arguments()
definition = InputDefinition()
definition.add_arguments([self.foo])
self.assertEqual([self.foo], definition.get_arguments())
definition.add_arguments([self.bar])
self.assertEqual([self.foo, self.bar], definition.get_arguments())
def test_has_argument(self):
self.initialize_arguments()
definition = InputDefinition()
definition.add_arguments([self.foo])
self.assertTrue(definition.has_argument('foo'))
self.assertFalse(definition.has_argument('bar'))
def test_get_invalid_argument(self):
self.initialize_arguments()
definition = InputDefinition()
definition.add_arguments([self.foo])
self.assertRaises(Exception, 'The "bar" argument does not exist.', definition.get_argument, "bar")
def test_get_argument(self):
self.initialize_arguments()
definition = InputDefinition()
definition.add_arguments([self.foo])
self.assertEqual(self.foo, definition.get_argument("foo"))
def test_get_argument(self):
self.initialize_arguments()
definition = InputDefinition()
definition.add_arguments([self.foo])
self.assertEqual(self.foo, definition.get_argument('foo'))
