From 008a780a9dc001c40a1bb246256bd17d93987a15 Mon Sep 17 00:00:00 2001
From: Matthias Volk <matthias.volk@cs.rwth-aachen.de>
Date: Fri, 23 Dec 2016 11:43:25 +0100
Subject: [PATCH] Added test cases for BitVector

---
 tests/storage/test_bitvector.py | 41 +++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 tests/storage/test_bitvector.py

diff --git a/tests/storage/test_bitvector.py b/tests/storage/test_bitvector.py
new file mode 100644
index 0000000..ea58fad
--- /dev/null
+++ b/tests/storage/test_bitvector.py
@@ -0,0 +1,41 @@
+import stormpy
+
+class TestBitvector:
+    def test_init_default(self):
+        bit = stormpy.BitVector()
+        assert bit.size() == 0
+        assert bit.number_of_set_bits() == 0
+    
+    def test_init_length(self):
+        bit = stormpy.BitVector(10)
+        assert bit.size() == 10
+        assert bit.number_of_set_bits() == 0
+        bit = stormpy.BitVector(5, True)
+        assert bit.size() == 5
+        assert bit.number_of_set_bits() == 5
+    
+    def test_init_vector(self):
+        bit = stormpy.BitVector(5, [2, 3])
+        assert bit.size() == 5
+        assert bit.number_of_set_bits() == 2
+        assert bit.get(0) == False
+        assert bit.get(1) == False
+        assert bit.get(2) == True
+        assert bit.get(3) == True
+        assert bit.get(4) == False
+    
+    def test_init_bitvector(self):
+        bit = stormpy.BitVector(7, [0, 6])
+        bit2 = stormpy.BitVector(bit)
+        assert bit == bit2
+        assert bit2.get(0) == True
+        assert bit2.get(6) == True
+    
+    def test_negate(self):
+        bit = stormpy.BitVector(7, [0, 6])
+        bit2 = stormpy.BitVector(bit)
+        bit = ~bit
+        assert bit.get(0) == False
+        assert bit.get(6) == False
+        for i in range(bit.size()):
+            assert bit.get(i) is not bit2.get(i)