From fe9d1ecbbe830f2e705e7baac726a4edf397dbd8 Mon Sep 17 00:00:00 2001
From: mjalalzai <33738574+MForensic@users.noreply.github.com>
Date: Sun, 11 Jun 2023 13:52:08 -0700
Subject: [PATCH] Creating a bitarray

---
 carnot/test_attestation.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

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)