Adding bool numpy test.

huggingface · Aug 17, 2023 · 2b0a49a · 2b0a49a
1 parent 0398834
commit 2b0a49a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/bindings/python/tests/test_simple.py b/bindings/python/tests/test_simple.py
@@ -198,6 +198,19 @@ def test_numpy_example(self):
         loaded = load(out)
         self.assertTensorEqual(tensors, loaded, np.allclose)
 
+    def test_numpy_bool(self):
+        tensors = {"a": np.asarray(False)}
+
+        save_file(tensors, "./out_bool.safetensors")
+        out = save(tensors)
+
+        # Now loading
+        loaded = load_file("./out_bool.safetensors")
+        self.assertTensorEqual(tensors, loaded, np.allclose)
+
+        loaded = load(out)
+        self.assertTensorEqual(tensors, loaded, np.allclose)
+
     def test_torch_example(self):
         tensors = {
             "a": torch.zeros((2, 2)),