def test_unequal_env_by_name_and_choices(self):
self.assertUnequalCominationPairs([
enum_by_name('BLA', ['SINGLETON']),
enum_by_name('BLUB', ['FOO', 'BAR']),
enum_by_name('BLA', ['FOO', 'BAR']),
enum_by_name('BLUB', ['ARG1', 'ARG2'])
])
def test_unequal_env_by_name(self):
self.assertUnequalCominationPairs([
enum_by_name("FOO", ['SINGLETON']),
enum_by_name("BAR", ['SINGLETON']),
enum_by_name("MY_VARIABLE", ['SINGLETON']),
enum_by_name("SOME_OTHER_VARIABLE", ['SINGLETON'])
])
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_3(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=None))
def test_invalid_enum_11(self):
with self.assertRaises(TypeError):
enum_by_name('FOO', 1)
def test_invalid_enum_10(self):
with self.assertRaises(ValueError):
enum_by_name('FOO', [])
def test_invalid_enum_5(self):
with self.assertRaises(TypeError):
enum_by_name(None, ['SINGLETON'])
def test_equals_3(self):
self.assertEqual(enum_by_name("BAZ", ['SINGLETON']),
enum_by_name("BAZ", ['SINGLETON']))
def test_valid_environment_variable_4(self):
enum_by_name('TEST_THIS_URL', ["SINGLETON"])
def setUp(self):
self.enum1 = enum_by_name('FOO', ['VALUE1', 'VALUE2'])
self.enum2 = enum_by_name('BAR', ['SINGLETON'])
def test_check_true_2(self):
self.assertThat(
enum_by_name('FOO_BAR', ['value1', 'value2']).state().is_satisfied)
def test_hash_3(self):
self.assertIsEqual(enum_by_name('FOO', choices=['SINGLETON']),
enum_by_name('FOO', choices=['SINGLETON']))
def test_equals_5(self):
self.assertEqual(enum_by_name("SOME_OTHER_VARIABLE", ['SINGLETON']),
enum_by_name("SOME_OTHER_VARIABLE", ['SINGLETON']))
def test_equals_4(self):
self.assertEqual(enum_by_name("MY_VARIABLE", ['SINGLETON']),
enum_by_name("MY_VARIABLE", ['SINGLETON']))
def assert_invalid_env_name(self, name):
self.assertValueError(lambda: enum_by_name(name, ["SINGLETON"]))
def test_valid_environment_variable_1(self):
enum_by_name('MY_URL', ["SINGLETON"])
def test_check_false_when_env_var_is_missing(self):
self.assertThatNot(
enum_by_name('FOO_BAR', ['value']).state().is_satisfied)
def test_valid_environment_variable_3(self):
enum_by_name('FOO', ["SINGLETON"])
def test_check_false_when_value_is_not_allowed(self):
self.assertThatNot(
enum_by_name('FOO_BAR', ['value2']).state().is_satisfied)
def test_get_value_2(self):
with patch.dict('os.environ', {'BAZ': 'value2'}):
self.assertEqual(
enum_by_name('BAZ', ['value2']).get_value(), 'value2')
def test_invalid_enum_4(self):
with self.assertRaises(ValueError):
enum_by_name('akfgus', ['SINGLETON'])
def requirements(self):
return Require(enum_by_name('FOO', choices=['VALUE1', 'VALUE2']))
def test_equals_1(self):
self.assertEqual(enum_by_name("FOO", ['SINGLETON']),
enum_by_name("FOO", ['SINGLETON']))
