def test_datauri_to_blob(): doc = Document(uri=os.path.join(cur_dir, 'test.png')) doc.convert_uri_to_datauri() doc.convert_image_datauri_to_blob() assert isinstance(doc.blob, np.ndarray) assert doc.mime_type == 'image/png' assert doc.blob.shape == (85, 152, 3) # h,w,c
def test_convert_uri_to_data_uri(uri, mimetype): doc = Document(uri=uri, mime_type=mimetype) doc.convert_uri_to_datauri() assert doc.uri.startswith(f'data:{mimetype}') assert doc.mime_type == mimetype
def test_datauri_to_blob(): doc = Document(uri=os.path.join(cur_dir, 'test.png')) doc.convert_uri_to_datauri() assert not doc.blob assert doc.mime_type == 'image/png'