def __init__(self,
models=None,
data=None,
labels=[],
repo_info=RepoInfo()):
super(TestDefinition, self).__init__(repo_info)
self.repo_info.category = MLObjectType.TEST_DEFINITION
self.models = models
self.data = data
self.labels = None
def __init__(self,
model,
data,
test_definition_version=LAST_VERSION,
model_version=LAST_VERSION,
data_version=LAST_VERSION,
repo_info=RepoInfo()):
super(RegressionTest, self).__init__(model,
data,
test_definition_version,
model_version,
data_version,
repo_info=repo_info)
def test_repo_info(self):
name = 'dummy'
version = '1.0.2'
repo_info = RepoInfo({'version': version, 'NAME': name})
self.assertEqual(repo_info.name, 'dummy')
self.assertEqual(repo_info['name'], 'dummy')
self.assertEqual(repo_info['NAME'], 'dummy')
self.assertEqual(repo_info[RepoInfoKey.NAME], 'dummy')
repo_info.set_fields({RepoInfoKey.BIG_OBJECTS: {'dummy': 'dummy2'}})
self.assertEqual(repo_info.name, 'dummy')
self.assertEqual(repo_info.big_objects['dummy'], 'dummy2')
self.assertEqual(repo_info.big_objects['dummy'], 'dummy2')
repo_info[RepoInfoKey.NAME] = 'dummy2'
self.assertEqual(repo_info.name, 'dummy2')
repo_info['NAME'] = 'dummy2'
self.assertEqual(repo_info['NAME'], 'dummy2')
repo_info['name'] = 'dummy2'
self.assertEqual(repo_info['NAME'], 'dummy2')
self.assertEqual(repo_info['dummy'], None)
repo_info_dict = repo_info.get_dictionary()
self.assertEqual(repo_info_dict['name'], 'dummy2')
# now test if dictionary contains only values for RepoInfoKeys
for k in RepoInfoKey:
self.assertEqual(repo_info_dict[k.value], repo_info[k])
def __init__(self,
model,
data,
test_definition_version=LAST_VERSION,
model_version=LAST_VERSION,
data_version=LAST_VERSION,
repo_info=RepoInfo()):
super(Test, self).__init__(repo_info)
self.test_definition = None
self.test_definition_version = test_definition_version
self.model_version = model_version
self.model = model
self.data_version = data_version
self.data = data
self.result = 'not run'
self.details = {}
def __init__(self,
reference='prod',
models=None,
data=None,
labels=None,
measures=None,
tol=1e-3,
repo_info=RepoInfo(),
relative=False):
super(RegressionTestDefinition, self).__init__(models,
data,
labels,
repo_info=repo_info)
self.measures = measures
self.reference = reference
self.tol = tol
self.repo_info.category = MLObjectType.TEST_DEFINITION
self.relative = relative
def __init__(self, a, b, repo_info=RepoInfo()):
super(TestClass, self).__init__(repo_info)
self.a = a
self._b = b