Skip to content

Commit b7d14d4

Browse files
committed
[PTDT-0] Update tests to contain classifications field for TextAnswer
1 parent 84d5fae commit b7d14d4

File tree

1 file changed

+70
-10
lines changed

1 file changed

+70
-10
lines changed

libs/labelbox/tests/data/annotation_import/conftest.py

Lines changed: 70 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2071,7 +2071,10 @@ def expected_export_v2_image():
20712071
{
20722072
"name": "text",
20732073
"value": "text",
2074-
"text_answer": {"content": "free form text..."},
2074+
"text_answer": {
2075+
"content": "free form text...",
2076+
"classifications": []
2077+
},
20752078
},
20762079
],
20772080
"relationships": [],
@@ -2103,7 +2106,10 @@ def expected_export_v2_audio():
21032106
{
21042107
"name": "text",
21052108
"value": "text",
2106-
"text_answer": {"content": "free form text..."},
2109+
"text_answer": {
2110+
"content": "free form text...",
2111+
"classifications": []
2112+
},
21072113
},
21082114
],
21092115
"segments": {},
@@ -2120,7 +2126,10 @@ def expected_export_v2_html():
21202126
{
21212127
"name": "text",
21222128
"value": "text",
2123-
"text_answer": {"content": "free form text..."},
2129+
"text_answer": {
2130+
"content": "free form text...",
2131+
"classifications": []
2132+
},
21242133
},
21252134
{
21262135
"name": "checklist",
@@ -2180,7 +2189,10 @@ def expected_export_v2_text():
21802189
{
21812190
"name": "text",
21822191
"value": "text",
2183-
"text_answer": {"content": "free form text..."},
2192+
"text_answer": {
2193+
"content": "free form text...",
2194+
"classifications": []
2195+
},
21842196
},
21852197
],
21862198
"relationships": [],
@@ -2328,7 +2340,40 @@ def expected_export_v2_document():
23282340
{
23292341
"name": "text",
23302342
"value": "text",
2331-
"text_answer": {"content": "free form text..."},
2343+
"text_answer": {
2344+
"content": "free form text...",
2345+
"classifications": []
2346+
},
2347+
},
2348+
],
2349+
"relationships": [],
2350+
}
2351+
return expected_annotations
2352+
2353+
2354+
@pytest.fixture()
2355+
def expected_export_v2_llm_prompt_creation():
2356+
expected_annotations = {
2357+
"objects": [],
2358+
"classifications": [
2359+
{
2360+
"name":
2361+
"checklist",
2362+
"value":
2363+
"checklist",
2364+
"checklist_answers": [{
2365+
"name": "option1",
2366+
"value": "option1",
2367+
"classifications": []
2368+
}],
2369+
},
2370+
{
2371+
"name": "text",
2372+
"value": "text",
2373+
"text_answer": {
2374+
"content": "free form text...",
2375+
"classifications": []
2376+
},
23322377
},
23332378
],
23342379
"relationships": [],
@@ -2344,11 +2389,17 @@ def expected_export_v2_llm_prompt_response_creation():
23442389
{
23452390
"name": "prompt-text",
23462391
"value": "prompt-text",
2347-
"text_answer": {"content": "free form text..."},
2392+
"text_answer": {
2393+
"content": "free form text...",
2394+
"classifications": []
2395+
},
23482396
},
23492397
{
23502398
"name": "response-text",
2351-
"text_answer": {"content": "free form text..."},
2399+
"text_answer": {
2400+
"content": "free form text...",
2401+
"classifications": []
2402+
},
23522403
"value": "response-text",
23532404
},
23542405
{
@@ -2390,7 +2441,10 @@ def expected_export_v2_llm_prompt_creation():
23902441
{
23912442
"name": "prompt-text",
23922443
"value": "prompt-text",
2393-
"text_answer": {"content": "free form text..."},
2444+
"text_answer": {
2445+
"content": "free form text...",
2446+
"classifications": []
2447+
},
23942448
},
23952449
],
23962450
"relationships": [],
@@ -2406,7 +2460,10 @@ def expected_export_v2_llm_response_creation():
24062460
"classifications": [
24072461
{
24082462
"name": "response-text",
2409-
"text_answer": {"content": "free form text..."},
2463+
"text_answer": {
2464+
"content": "free form text...",
2465+
"classifications": []
2466+
},
24102467
"value": "response-text",
24112468
},
24122469
{
@@ -2518,7 +2575,10 @@ def expected_exports_v2_mmc(mmc_example_data_row_message_ids):
25182575
{
25192576
"name": "text",
25202577
"value": "text",
2521-
"text_answer": {"content": "free form text..."},
2578+
"text_answer": {
2579+
"content": "free form text...",
2580+
"classifications": []
2581+
},
25222582
},
25232583
{
25242584
"name": "radio_index",

0 commit comments

Comments
 (0)