def test_get_argument_required_count(self): self.initialize_arguments() definition = InputDefinition() definition.add_argument(self.foo2) self.assertEqual(1, definition.get_argument_required_count()) definition.add_argument(self.foo) self.assertEqual(1, definition.get_argument_required_count())