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