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)
