FilesExpand file tree

 main

/

test_json.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

35 lines (29 loc) · 996 Bytes

 main

/

test_json.py

Copy path

Top

File metadata and controls

• Code
• Blame

35 lines (29 loc) · 996 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

import io

import pytest

from replicate.json import base64_encode_file

@pytest.mark.parametrize(

"content, filename, expected",

[

(b"Hello, World!", "test.txt", "data:text/plain;base64,SGVsbG8sIFdvcmxkIQ=="),

(b"\x89PNG\r\n\x1a\n", "image.png", "data:image/png;base64,iVBORw0KGgo="),

(

"{'key': 'value'}",

"data.json",

"data:application/json;base64,eydrZXknOiAndmFsdWUnfQ==",

),

(

b"Random bytes",

None,

"data:application/octet-stream;base64,UmFuZG9tIGJ5dGVz",

),

],

)

def test_base64_encode_file(content, filename, expected):

# Create a file-like object with the given content

file = io.BytesIO(content if isinstance(content, bytes) else content.encode())

# Set the filename if provided

if filename:

file.name = filename

# Call the function and check the result

result = base64_encode_file(file)

assert result == expected