diff --git a/carnot/test_attestation.py b/carnot/test_attestation.py index 73131b0..d9ae883 100644 --- a/carnot/test_attestation.py +++ b/carnot/test_attestation.py @@ -15,3 +15,38 @@ class TestCountOnBitarrayFields(unittest.TestCase): self.assertEqual(result, expected_result) + +class TestCreateCommitteeBitArray(unittest.TestCase): + class Vote: + def __init__(self, voter): + self.voter = voter + + def test_createCommitteeBitArray_with_smaller_committee_size(self): + voters = [ + self.Vote("Alice"), + self.Vote("Bob"), + self.Vote("Charlie") + ] + committee_size = 2 + + try: + result = createCommitteeBitArray(voters, committee_size) + self.fail("AssertionError should have been raised.") + except AssertionError: + pass + + + def test_createCommitteeBitArray_with_larger_committee_size(self): + voters = [ + self.Vote("Alice"), + self.Vote("Bob"), + self.Vote("Charlie"), + self.Vote("Dave"), + self.Vote("Eve") + ] + committee_size = 6 + + result = createCommitteeBitArray(voters, committee_size) + expected_result = [True, True, True, True, True, False] + + self.assertEqual(result, expected_result)