static trait unit tests

This commit is contained in:
Tegiminis 2022-02-12 16:51:27 -08:00
parent 7510229edb
commit 321d4005fc

View file

@ -274,13 +274,14 @@ class TestTraitStatic(_TraitHandlerBase):
trait_type="static",
base=1,
mod=2,
mult=2.0,
extra_val1="xvalue1",
extra_val2="xvalue2",
)
self.trait = self.traithandler.get("test1")
def _get_values(self):
return self.trait.base, self.trait.mod, self.trait.value
return self.trait.base, self.trait.mod, self.trait.mult, self.trait.value
def test_init(self):
self.assertEqual(
@ -290,25 +291,33 @@ class TestTraitStatic(_TraitHandlerBase):
"trait_type": "static",
"base": 1,
"mod": 2,
"mult": 2.0,
"extra_val1": "xvalue1",
"extra_val2": "xvalue2",
},
)
def test_value(self):
"""value is base + mod"""
self.assertEqual(self._get_values(), (1, 2, 3))
"""value is (base + mod) * mult"""
self.assertEqual(self._get_values(), (1, 2, 2.0, 6))
self.trait.base += 4
self.assertEqual(self._get_values(), (5, 2, 7))
self.assertEqual(self._get_values(), (5, 2, 2.0, 14))
self.trait.mod -= 1
self.assertEqual(self._get_values(), (5, 1, 6))
self.assertEqual(self._get_values(), (5, 1, 2.0, 12))
self.trait.mult += 1.0
self.assertEqual(self._get_values(), (5, 1, 3.0, 18))
self.trait.mult = 0.75
self.assertEqual(self._get_values(), (5, 1, 0.75, 4.5))
def test_delete(self):
"""Deleting resets to default."""
del self.trait.base
self.assertEqual(self._get_values(), (0, 2, 2))
self.assertEqual(self._get_values(), (0, 2, 2.0, 4))
del self.trait.mult
self.assertEqual(self._get_values(), (0, 2, 1.0, 2))
del self.trait.mod
self.assertEqual(self._get_values(), (0, 0, 0))
self.assertEqual(self._get_values(), (0, 0, 1.0, 0))
class TestTraitCounter(_TraitHandlerBase):