def test_input_schema_fit(self): self.maxDiff = None self.assertEqual( LogisticRegression.input_schema_fit(), LogisticRegression.get_schema("input_fit"), ) self.assertEqual( (NMF >> LogisticRegression).input_schema_fit(), NMF.get_schema("input_fit") ) self.assertEqual( IdentityWrapper(op=LogisticRegression).input_schema_fit(), LogisticRegression.get_schema("input_fit"), ) actual = (TfidfVectorizer | NMF).input_schema_fit() expected = { "anyOf": [ { "type": "object", "required": ["X"], "additionalProperties": False, "properties": { "X": { "anyOf": [ {"type": "array", "items": {"type": "string"}}, { "type": "array", "items": { "type": "array", "minItems": 1, "maxItems": 1, "items": {"type": "string"}, }, }, ] }, "y": {}, }, }, { "type": "object", "required": ["X"], "additionalProperties": False, "properties": { "X": { "type": "array", "items": { "type": "array", "items": {"type": "number", "minimum": 0.0}, }, }, "y": {}, }, }, ] } self.assertEqual(actual, expected)
def test_input_schema_fit(self): self.maxDiff = None self.assertEqual(LogisticRegression.input_schema_fit(), LogisticRegression.get_schema('input_fit')) self.assertEqual((NMF >> LogisticRegression).input_schema_fit(), NMF.get_schema('input_fit')) self.assertEqual( IdentityWrapper(op=LogisticRegression).input_schema_fit(), LogisticRegression.get_schema('input_fit')) actual = (TfidfVectorizer | NMF).input_schema_fit() expected = { 'anyOf': [{ 'type': 'object', 'required': ['X'], 'additionalProperties': False, 'properties': { 'X': { 'anyOf': [{ 'type': 'array', 'items': { 'type': 'string' } }, { 'type': 'array', 'items': { 'type': 'array', 'minItems': 1, 'maxItems': 1, 'items': { 'type': 'string' } } }] }, 'y': {} } }, { 'type': 'object', 'required': ['X'], 'additionalProperties': False, 'properties': { 'X': { 'type': 'array', 'items': { 'type': 'array', 'items': { 'type': 'number', 'minimum': 0.0 } } }, 'y': {} } }] } self.assertEqual(actual, expected)