Added more tests

Author: ankane

tests/test_bit.py

@@ -96,6 +96,23 @@ def test_eq(self) -> None:
         assert Bit([True, False, True]) != Bit([True, False, False])
         assert Bit([True, False, True]) != 1
 
+    def test_to_list(self) -> None:
+        assert Bit('101').to_list() == [True, False, True]
+
+    def test_to_numpy(self) -> None:
+        if np is None:
+            pytest.skip('NumPy required')
+
+        arr = Bit('101').to_numpy()
+        assert np.array_equal(arr, np.array([True, False, True]))
+        assert arr.dtype == np.bool_
+
+    def test_to_text(self) -> None:
+        assert Bit('101').to_text() == '101'
+
+    def test_to_binary(self) -> None:
+        assert Bit('101').to_binary() == struct.pack('>iB', 3, 5 << 5)
+
     def test_from_text(self) -> None:
         assert Bit.from_text('101') == Bit('101')
 

tests/test_half_vector.py

@@ -60,6 +60,23 @@ def test_eq(self) -> None:
     def test_dimensions(self) -> None:
         assert HalfVector([1, 2, 3]).dimensions() == 3
 
+    def test_to_list(self) -> None:
+        assert HalfVector([1.5, 2, 3]).to_list() == [1.5, 2, 3]
+
+    def test_to_numpy(self) -> None:
+        if np is None:
+            pytest.skip('NumPy required')
+
+        arr = HalfVector([1.5, 2, 3]).to_numpy()
+        assert np.array_equal(arr, np.array([1.5, 2, 3]))
+        assert arr.dtype == np.float16
+
+    def test_to_text(self) -> None:
+        assert HalfVector([1.5, 2, 3]).to_text() == '[1.5,2.0,3.0]'
+
+    def test_to_binary(self) -> None:
+        assert HalfVector([1.5, 2, 3]).to_binary() == struct.pack('>hh3e', 3, 0, 1.5, 2, 3)
+
     def test_from_text(self) -> None:
         assert HalfVector.from_text('[1.5,2,3]') == HalfVector([1.5, 2, 3])
 

tests/test_sparse_vector.py

@@ -155,9 +155,22 @@ def test_to_coo(self) -> None:
 
         assert np.array_equal(SparseVector([1, 0, 2, 0, 3, 0]).to_coo().toarray(), [[1, 0, 2, 0, 3, 0]])
 
-    def test_zero_vector_text(self) -> None:
-        vec = SparseVector({}, 3)
-        assert vec.to_list() == SparseVector.from_text(vec.to_text()).to_list()
+    def test_to_list(self) -> None:
+        assert SparseVector({2: 2, 4: 3, 0: 1.5, 3: 0}, 6).to_list() == [1.5, 0, 2, 0, 3, 0]
+
+    def test_to_numpy(self) -> None:
+        if np is None:
+            pytest.skip('NumPy required')
+
+        arr = SparseVector({2: 2, 4: 3, 0: 1.5, 3: 0}, 6).to_numpy()
+        assert np.array_equal(arr, np.array([1.5, 0, 2, 0, 3, 0]))
+        assert arr.dtype == np.float32
+
+    def test_to_text(self) -> None:
+        assert SparseVector({2: 2, 4: 3, 0: 1.5, 3: 0}, 6).to_text() == '{1:1.5,3:2.0,5:3.0}/6'
+
+    def test_to_binary(self) -> None:
+        assert SparseVector({2: 2, 4: 3, 0: 1.5, 3: 0}, 6).to_binary() == struct.pack('>iii3i3f', 6, 3, 0, 0, 2, 4, 1.5, 2, 3)
 
     def test_from_text(self) -> None:
         assert SparseVector.from_text('{1:1.5,3:2,5:3}/6') == SparseVector([1.5, 0, 2, 0, 3, 0])

tests/test_vector.py

@@ -60,6 +60,23 @@ def test_eq(self) -> None:
     def test_dimensions(self) -> None:
         assert Vector([1, 2, 3]).dimensions() == 3
 
+    def test_to_list(self) -> None:
+        assert Vector([1.5, 2, 3]).to_list() == [1.5, 2, 3]
+
+    def test_to_numpy(self) -> None:
+        if np is None:
+            pytest.skip('NumPy required')
+
+        arr = Vector([1.5, 2, 3]).to_numpy()
+        assert np.array_equal(arr, np.array([1.5, 2, 3]))
+        assert arr.dtype == np.float32
+
+    def test_to_text(self) -> None:
+        assert Vector([1.5, 2, 3]).to_text() == '[1.5,2.0,3.0]'
+
+    def test_to_binary(self) -> None:
+        assert Vector([1.5, 2, 3]).to_binary() == struct.pack('>hh3f', 3, 0, 1.5, 2, 3)
+
     def test_from_text(self) -> None:
         assert Vector.from_text('[1.5,2,3]') == Vector([1.5, 2, 3])