def test_step_should_embed_data_without_encoding(): """A Step should be able to embed data without an encoding""" # given step = Step(1, "keyword", "used_keyword", "text", None, None, None, None) # when step.embed("data", encode_data_to_base64=False) # then assert step.embeddings == [{"data": "data", "mime_type": "text/plain"}]
def test_step_should_embed_data_with_base64_encoding(): """A Step should be able to embed data base64 encoded""" # given step = Step(1, "keyword", "used_keyword", "text", None, None, None, None) data_b64_encoded = "ZGF0YQ==" # == "data" # when step.embed("data", encode_data_to_base64=True) # then assert step.embeddings == [{"data": data_b64_encoded, "mime_type": "text/plain"}]