chaintool/shep
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add mask

Browse Source
This commit is contained in:
lash
2022-03-18 18:34:49 +00:00
parent 6680b897b3
commit 2beeb4c725
3 changed files with 36 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tests/test_state.py
View File

@@ -216,5 +216,26 @@ class TestState(unittest.TestCase):
states.alias('baz', states.FOO | states.BAR)
def test_mask(self):
states = State(3)
states.add('foo')
states.add('bar')
states.add('baz')
states.alias('all', states.FOO | states.BAR | states.BAZ)
mask = states.mask('xyzzy', states.FOO | states.BAZ)
self.assertEqual(mask, states.BAR)
def test_mask_dynamic(self):
states = State(0)
states.add('foo')
states.add('bar')
states.add('baz')
states.alias('all', states.FOO | states.BAR | states.BAZ)
mask = states.mask('xyzzy', states.FOO | states.BAZ)
self.assertEqual(mask, states.BAR)
if __name__ == '__main__':
unittest.main()