def setUp(self):
self.maxDiff = None
self.td1 = TrainDescription(
train_name='test_train',
train_version='1.0',
properties={
1: url_by_name('FOO'),
2: token_by_name('BAR'),
3: enum_by_name('BAZ', choices=['VALUE1', 'VALUE2'])
},
formulas=[ConjunctiveNormalForm(Clause(1))],
model_summary='model summary',
algorithm_requirement=FormulaAlgorithmRequirement(1))
self.td2 = TrainDescription(
train_name='test_train',
train_version='1.0',
properties={
1: url_by_name('FOO'),
2: token_by_name('BAR'),
3: enum_by_name('BAZ', choices=['VALUE1', 'VALUE2'])
},
formulas=[ConjunctiveNormalForm(Clause(1))],
model_summary='model summary',
algorithm_requirement=None)
self.td3 = TrainDescription(
train_name='test_train',
train_version='1.0',
properties={
1: url_by_name('FOO'),
2: token_by_name('BAR'),
3: enum_by_name('BAZ', choices=['VALUE1', 'VALUE2'])
},
formulas=[
ConjunctiveNormalForm(
Clause(1), Clause(-2, 1, 3), Clause(3, 1, -2, -3)
),
ConjunctiveNormalForm(
Clause(3, 1), Clause(2, 1, 3, -3)
),
ConjunctiveNormalForm(
Clause(-1)
)
],
model_summary='model summary',
algorithm_requirement=FormulaAlgorithmRequirement(3))
def test_invalid_train_description_2(self):
self.assertValueError(lambda: TrainDescription(
train_name='foo',
train_version='1.0',
properties={
1: url_by_name('FOO'),
2: token_by_name('BAR'),
3: enum_by_name('BAZ', choices=['VALUE1', 'VALUE2'])
},
formulas=[],
model_summary='model summary',
algorithm_requirement=FormulaAlgorithmRequirement(-1)))
def describe(self, info: StationRuntimeInfo) -> TrainDescription:
requirements = self.requirements()
if requirements is not None:
properties = requirements.props
formulas = [requirements.cnf()]
alg_req = FormulaAlgorithmRequirement(1)
else:
properties = {}
formulas = []
alg_req = None
return TrainDescription(train_name=self.train_name,
train_version=self.train_version,
properties=properties,
formulas=formulas,
model_summary=self.model_summary(),
algorithm_requirement=alg_req)
def test_eq_hash_4(self):
self.assertIsEqual(FormulaAlgorithmRequirement(127), self.alg4)
def test_eq_hash_3(self):
self.assertIsEqual(FormulaAlgorithmRequirement(3), self.alg3)
def setUp(self):
self.alg1 = FormulaAlgorithmRequirement(1)
self.alg2 = FormulaAlgorithmRequirement(2)
self.alg3 = FormulaAlgorithmRequirement(3)
self.alg4 = FormulaAlgorithmRequirement(127)
def test_type_error_6(self):
self.assertTypeError(lambda: FormulaAlgorithmRequirement([]))
class FormulaAlgorithmRequirementTests(BaseTest):
def setUp(self):
self.alg1 = FormulaAlgorithmRequirement(1)
self.alg2 = FormulaAlgorithmRequirement(2)
self.alg3 = FormulaAlgorithmRequirement(3)
self.alg4 = FormulaAlgorithmRequirement(127)
################################################################################
# Value Error
################################################################################
def test_value_error_1(self):
self.assertValueError(lambda: FormulaAlgorithmRequirement(0))
def test_value_error_2(self):
self.assertValueError(lambda: FormulaAlgorithmRequirement(-1))
def test_value_error_3(self):
self.assertValueError(lambda: FormulaAlgorithmRequirement(-10))
################################################################################
# Type Error
################################################################################
def test_type_error_1(self):
self.assertTypeError(lambda: FormulaAlgorithmRequirement(None))
def test_type_error_2(self):
self.assertTypeError(lambda: FormulaAlgorithmRequirement(True))
def test_type_error_3(self):
self.assertTypeError(lambda: FormulaAlgorithmRequirement(False))
def test_type_error_4(self):
self.assertTypeError(lambda: FormulaAlgorithmRequirement('foo'))
def test_type_error_5(self):
self.assertTypeError(lambda: FormulaAlgorithmRequirement({}))
def test_type_error_6(self):
self.assertTypeError(lambda: FormulaAlgorithmRequirement([]))
################################################################################
# Equal and hash
################################################################################
def test_eq_hash_1(self):
self.assertIsEqual(FormulaAlgorithmRequirement(1), self.alg1)
def test_eq_hash_2(self):
self.assertIsEqual(FormulaAlgorithmRequirement(2), self.alg2)
def test_eq_hash_3(self):
self.assertIsEqual(FormulaAlgorithmRequirement(3), self.alg3)
def test_eq_hash_4(self):
self.assertIsEqual(FormulaAlgorithmRequirement(127), self.alg4)
def test_unequal_combination_pairs(self):
self.assertUnequalCominationPairs(
[self.alg1, self.alg2, self.alg3, self.alg4])
################################################################################
# Copy
################################################################################
def test_copy_1(self):
self.assertCopiesAreEqualOf(self.alg1)
def test_copy_2(self):
self.assertCopiesAreEqualOf(self.alg2)
def test_copy_3(self):
self.assertCopiesAreEqualOf(self.alg3)
def test_copy_4(self):
self.assertCopiesAreEqualOf(self.alg4)
################################################################################
# As Dict
################################################################################
def test_as_dict_1(self):
self.checkExpect(expect={
'formula_id':
1,
'@type': ['FormulaAlgorithmRequirement', 'AlgorithmRequirement'],
'@typeName':
'FormulaAlgorithmRequirement',
"@typeSystem": {
'name': 'pythonclass',
'version': '1.0'
},
},
actual=self.alg1.as_simple_mapping())
def test_as_dict_2(self):
self.checkExpect(expect={
'formula_id':
2,
'@type': ['FormulaAlgorithmRequirement', 'AlgorithmRequirement'],
'@typeName':
'FormulaAlgorithmRequirement',
"@typeSystem": {
'name': 'pythonclass',
'version': '1.0'
},
},
actual=self.alg2.as_simple_mapping())
def test_as_dict_3(self):
self.checkExpect(expect={
'formula_id':
3,
'@type': ['FormulaAlgorithmRequirement', 'AlgorithmRequirement'],
'@typeName':
'FormulaAlgorithmRequirement',
"@typeSystem": {
'name': 'pythonclass',
'version': '1.0'
},
},
actual=self.alg3.as_simple_mapping())
def test_as_dict_4(self):
self.checkExpect(expect={
'formula_id':
127,
'@type': ['FormulaAlgorithmRequirement', 'AlgorithmRequirement'],
'@typeName':
'FormulaAlgorithmRequirement',
"@typeSystem": {
'name': 'pythonclass',
'version': '1.0'
},
},
actual=self.alg4.as_simple_mapping())
################################################################################
# Type
################################################################################
def test_type_1(self):
self.checkExpect(
expect=['FormulaAlgorithmRequirement', 'AlgorithmRequirement'],
actual=self.alg1.type)
def test_type_2(self):
self.checkExpect(
expect=['FormulaAlgorithmRequirement', 'AlgorithmRequirement'],
actual=self.alg2.type)
def test_type_3(self):
self.checkExpect(
expect=['FormulaAlgorithmRequirement', 'AlgorithmRequirement'],
actual=self.alg3.type)
def test_type_4(self):
self.checkExpect(
expect=['FormulaAlgorithmRequirement', 'AlgorithmRequirement'],
actual=self.alg4.type)
################################################################################
# typeName
################################################################################
def test_type_name_1(self):
self.checkExpect(expect='FormulaAlgorithmRequirement',
actual=self.alg1.type_name)
def test_type_name_2(self):
self.checkExpect(expect='FormulaAlgorithmRequirement',
actual=self.alg2.type_name)
def test_type_name_3(self):
self.checkExpect(expect='FormulaAlgorithmRequirement',
actual=self.alg3.type_name)
def test_type_name_4(self):
self.checkExpect(expect='FormulaAlgorithmRequirement',
actual=self.alg4.type_name)
################################################################################
# Data
################################################################################
def test_data_1(self):
self.checkExpect(expect={'formula_id': 1}, actual=self.alg1.data)
def test_data_2(self):
self.checkExpect(expect={'formula_id': 2}, actual=self.alg2.data)
def test_data_3(self):
self.checkExpect(expect={'formula_id': 3}, actual=self.alg3.data)
def test_data_4(self):
self.checkExpect(expect={'formula_id': 127}, actual=self.alg4.data)
def test_value_error_3(self):
self.assertValueError(lambda: FormulaAlgorithmRequirement(-10))