Appease code cov

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()

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"))

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)

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"