Skip to content

Commit d75ce34

Browse files
author
Adrian Chang
committed
buffered result
1 parent 8bfe14c commit d75ce34

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

libs/labelbox/tests/data/export/streamable/test_export_data_rows_streamable.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,25 @@ def test_with_data_row_object_buffered(self, client, data_row,
4747
assert export_task.get_total_lines(stream_type=StreamType.RESULT) == 1
4848
assert list(export_task.get_buffered_stream())[0].json["data_row"]["id"] == data_row.uid
4949

50+
def test_with_data_row_object_buffered_result(self, client, data_row,
51+
wait_for_data_row_processing):
52+
data_row = wait_for_data_row_processing(client, data_row)
53+
time.sleep(7) # temp fix for ES indexing delay
54+
export_task = DataRow.export(
55+
client=client,
56+
data_rows=[data_row],
57+
task_name="TestExportDataRow:test_with_data_row_object_buffered_result",
58+
)
59+
export_task.wait_till_done()
60+
assert export_task.status == "COMPLETE"
61+
assert isinstance(export_task, ExportTask)
62+
assert export_task.has_result()
63+
assert export_task.has_errors() is False
64+
assert export_task.get_total_file_size(
65+
stream_type=StreamType.RESULT) > 0
66+
assert export_task.get_total_lines(stream_type=StreamType.RESULT) == 1
67+
assert export_task.result[0]["data_row"]["id"] == data_row.uid
68+
5069
def test_with_id(self, client, data_row, wait_for_data_row_processing):
5170
data_row = wait_for_data_row_processing(client, data_row)
5271
time.sleep(7) # temp fix for ES indexing delay

0 commit comments

Comments
 (0)