From 27c1701cb5050255c3300a428797a66c3e3b7e17 Mon Sep 17 00:00:00 2001
From: Collin Dutter <collin@griptape.ai>
Date: Tue, 10 Sep 2024 14:16:18 -0700
Subject: [PATCH] Appease code cov

---
 griptape/artifacts/boolean_artifact.py        |  2 +-
 tests/unit/artifacts/test_blob_artifact.py    | 10 ++++++++++
 tests/unit/artifacts/test_boolean_artifact.py | 11 +++++++++++
 tests/unit/artifacts/test_text_artifact.py    |  5 +++++
 4 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/griptape/artifacts/boolean_artifact.py b/griptape/artifacts/boolean_artifact.py
index dd8bb1504b..eb135824d4 100644
--- a/griptape/artifacts/boolean_artifact.py
+++ b/griptape/artifacts/boolean_artifact.py
@@ -34,7 +34,7 @@ def __add__(self, other: BaseArtifact) -> BooleanArtifact:
         raise ValueError("Cannot add BooleanArtifact with other artifacts")
 
     def __eq__(self, value: object) -> bool:
-        return self.value is value
+        return self.value == value
 
     def to_text(self) -> str:
         return str(self.value).lower()
diff --git a/tests/unit/artifacts/test_blob_artifact.py b/tests/unit/artifacts/test_blob_artifact.py
index d0f04b1e57..9db5e21f5d 100644
--- a/tests/unit/artifacts/test_blob_artifact.py
+++ b/tests/unit/artifacts/test_blob_artifact.py
@@ -12,6 +12,9 @@ def test_value_type_conversion(self):
     def test_to_text(self):
         assert BlobArtifact(b"foobar", name="foobar.txt").to_text() == "foobar"
 
+    def test_to_bytes(self):
+        assert BlobArtifact(b"foo").to_bytes() == b"foo"
+
     def test_to_text_encoding(self):
         assert (
             BlobArtifact("ß".encode("ascii", errors="backslashreplace"), name="foobar.txt", encoding="ascii").to_text()
@@ -50,6 +53,13 @@ def test_deserialization(self):
     def test_name(self):
         assert BlobArtifact(b"foo", name="bar").name == "bar"
 
+    def test_mime_type(self):
+        assert BlobArtifact(b"foo").mime_type == "application/octet-stream"
+
+    def test___add__(self):
+        with pytest.raises(TypeError):
+            BlobArtifact(b"foo") + BlobArtifact(b"bar")
+
     def test___bool__(self):
         assert not bool(BlobArtifact(b""))
         assert bool(BlobArtifact(b"foo"))
diff --git a/tests/unit/artifacts/test_boolean_artifact.py b/tests/unit/artifacts/test_boolean_artifact.py
index 57bbf16622..6ed21608d4 100644
--- a/tests/unit/artifacts/test_boolean_artifact.py
+++ b/tests/unit/artifacts/test_boolean_artifact.py
@@ -10,6 +10,7 @@ def test_parse_bool(self):
         assert BooleanArtifact.parse_bool("false").value is False
         assert BooleanArtifact.parse_bool("True").value is True
         assert BooleanArtifact.parse_bool("False").value is False
+        assert BooleanArtifact.parse_bool(True).value is True
 
         with pytest.raises(ValueError):
             BooleanArtifact.parse_bool("foo")
@@ -35,3 +36,13 @@ def test_value_type_conversion(self):
         assert BooleanArtifact([]).value is False
         assert BooleanArtifact(False).value is False
         assert BooleanArtifact(True).value is True
+
+    def test_to_text(self):
+        assert BooleanArtifact(True).to_text() == "true"
+        assert BooleanArtifact(False).to_text() == "false"
+
+    def test__eq__(self):
+        assert BooleanArtifact(True) == BooleanArtifact(True)
+        assert BooleanArtifact(False) == BooleanArtifact(False)
+        assert BooleanArtifact(True) != BooleanArtifact(False)
+        assert BooleanArtifact(False) != BooleanArtifact(True)
diff --git a/tests/unit/artifacts/test_text_artifact.py b/tests/unit/artifacts/test_text_artifact.py
index 9e00e2d2ed..dda256d27a 100644
--- a/tests/unit/artifacts/test_text_artifact.py
+++ b/tests/unit/artifacts/test_text_artifact.py
@@ -18,6 +18,11 @@ def test___add__(self):
     def test_to_dict(self):
         assert TextArtifact("foobar").to_dict()["value"] == "foobar"
 
+    def test_to_bytes(self):
+        artifact = TextArtifact("foobar")
+
+        assert artifact.to_bytes() == b"foobar"
+
     def test_from_dict(self):
         assert BaseArtifact.from_dict(TextArtifact("foobar").to_dict()).value == "foobar"