forked from a2aproject/a2a-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_parts.py
More file actions
184 lines (142 loc) · 4.34 KB
/
Copy pathtest_parts.py
File metadata and controls
184 lines (142 loc) · 4.34 KB
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
from a2a.types import (
DataPart,
FilePart,
FileWithBytes,
FileWithUri,
Part,
TextPart,
)
from a2a.utils.parts import (
get_data_parts,
get_file_parts,
get_text_parts,
)
class TestGetTextParts:
def test_get_text_parts_single_text_part(self):
# Setup
parts = [Part(root=TextPart(text='Hello world'))]
# Exercise
result = get_text_parts(parts)
# Verify
assert result == ['Hello world']
def test_get_text_parts_multiple_text_parts(self):
# Setup
parts = [
Part(root=TextPart(text='First part')),
Part(root=TextPart(text='Second part')),
Part(root=TextPart(text='Third part')),
]
# Exercise
result = get_text_parts(parts)
# Verify
assert result == ['First part', 'Second part', 'Third part']
def test_get_text_parts_empty_list(self):
# Setup
parts = []
# Exercise
result = get_text_parts(parts)
# Verify
assert result == []
class TestGetDataParts:
def test_get_data_parts_single_data_part(self):
# Setup
parts = [Part(root=DataPart(data={'key': 'value'}))]
# Exercise
result = get_data_parts(parts)
# Verify
assert result == [{'key': 'value'}]
def test_get_data_parts_multiple_data_parts(self):
# Setup
parts = [
Part(root=DataPart(data={'key1': 'value1'})),
Part(root=DataPart(data={'key2': 'value2'})),
]
# Exercise
result = get_data_parts(parts)
# Verify
assert result == [{'key1': 'value1'}, {'key2': 'value2'}]
def test_get_data_parts_mixed_parts(self):
# Setup
parts = [
Part(root=TextPart(text='some text')),
Part(root=DataPart(data={'key1': 'value1'})),
Part(root=DataPart(data={'key2': 'value2'})),
]
# Exercise
result = get_data_parts(parts)
# Verify
assert result == [{'key1': 'value1'}, {'key2': 'value2'}]
def test_get_data_parts_no_data_parts(self):
# Setup
parts = [
Part(root=TextPart(text='some text')),
]
# Exercise
result = get_data_parts(parts)
# Verify
assert result == []
def test_get_data_parts_empty_list(self):
# Setup
parts = []
# Exercise
result = get_data_parts(parts)
# Verify
assert result == []
class TestGetFileParts:
def test_get_file_parts_single_file_part(self):
# Setup
file_with_uri = FileWithUri(
uri='file://path/to/file', mimeType='text/plain'
)
parts = [Part(root=FilePart(file=file_with_uri))]
# Exercise
result = get_file_parts(parts)
# Verify
assert result == [file_with_uri]
def test_get_file_parts_multiple_file_parts(self):
# Setup
file_with_uri1 = FileWithUri(
uri='file://path/to/file1', mime_type='text/plain'
)
file_with_bytes = FileWithBytes(
bytes='ZmlsZSBjb250ZW50',
mime_type='application/octet-stream', # 'file content'
)
parts = [
Part(root=FilePart(file=file_with_uri1)),
Part(root=FilePart(file=file_with_bytes)),
]
# Exercise
result = get_file_parts(parts)
# Verify
assert result == [file_with_uri1, file_with_bytes]
def test_get_file_parts_mixed_parts(self):
# Setup
file_with_uri = FileWithUri(
uri='file://path/to/file', mime_type='text/plain'
)
parts = [
Part(root=TextPart(text='some text')),
Part(root=FilePart(file=file_with_uri)),
]
# Exercise
result = get_file_parts(parts)
# Verify
assert result == [file_with_uri]
def test_get_file_parts_no_file_parts(self):
# Setup
parts = [
Part(root=TextPart(text='some text')),
Part(root=DataPart(data={'key': 'value'})),
]
# Exercise
result = get_file_parts(parts)
# Verify
assert result == []
def test_get_file_parts_empty_list(self):
# Setup
parts = []
# Exercise
result = get_file_parts(parts)
# Verify
assert result == []