def test_invalid_comparison_mode(self):
trait = CTrait(TraitKind.trait)
# comparison modes other than {0,1,2}
# are invalid
with self.assertRaises(ValueError):
trait.comparison_mode = -1
with self.assertRaises(ValueError):
trait.comparison_mode = 3
def test_comparison_mode_unchanged_if_invalid(self):
trait = CTrait(TraitKind.trait)
default_comparison_mode = trait.comparison_mode
self.assertNotEqual(default_comparison_mode, ComparisonMode.none)
trait.comparison_mode = ComparisonMode.none
with self.assertRaises(ValueError):
trait.comparison_mode = -1
self.assertEqual(trait.comparison_mode, ComparisonMode.none)
def test_comparison_mode_enum(self):
trait = CTrait(TraitKind.trait)
trait.comparison_mode = ComparisonMode.none
self.assertIsInstance(trait.comparison_mode, ComparisonMode)
self.assertEqual(trait.comparison_mode, ComparisonMode.none)
trait.comparison_mode = ComparisonMode.identity
self.assertIsInstance(trait.comparison_mode, ComparisonMode)
self.assertEqual(trait.comparison_mode, ComparisonMode.identity)
trait.comparison_mode = ComparisonMode.equality
self.assertIsInstance(trait.comparison_mode, ComparisonMode)
self.assertEqual(trait.comparison_mode, ComparisonMode.equality)
