From c9a28604edc4478767755b16af6e4cafaa95b45b Mon Sep 17 00:00:00 2001
From: nnegrey <nnegrey@google.com>
Date: Tue, 11 Feb 2020 10:37:33 -0700
Subject: [PATCH] fix: datalabeling tests

---
 datalabeling/label_image_test.py | 9 ++++++---
 datalabeling/label_text_test.py  | 9 ++++++---
 datalabeling/label_video_test.py | 9 ++++++---
 3 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/datalabeling/label_image_test.py b/datalabeling/label_image_test.py
index 3917ef0481a..ddb74a03105 100644
--- a/datalabeling/label_image_test.py
+++ b/datalabeling/label_image_test.py
@@ -88,12 +88,15 @@ def instruction():
 # to be deleted before the annotation_spec_set can be deleted.
 @pytest.mark.slow
 def test_label_image(capsys, annotation_spec_set, instruction, dataset):
+    annotation_spec_set_name = annotation_spec_set.name
+    instruction_name = instruction.name
+    dataset_name = dataset.name
 
     # Start labeling.
     response = label_image.label_image(
-        dataset.name,
-        instruction.name,
-        annotation_spec_set.name
+        dataset_name,
+        instruction_name,
+        annotation_spec_set_name
     )
     out, _ = capsys.readouterr()
     assert 'Label_image operation name: ' in out
diff --git a/datalabeling/label_text_test.py b/datalabeling/label_text_test.py
index 4894c38304c..0e1f37d0c01 100644
--- a/datalabeling/label_text_test.py
+++ b/datalabeling/label_text_test.py
@@ -88,12 +88,15 @@ def instruction():
 # to be deleted before the annotation_spec_set can be deleted.
 @pytest.mark.slow
 def test_label_text(capsys, annotation_spec_set, instruction, dataset):
+    annotation_spec_set_name = annotation_spec_set.name
+    instruction_name = instruction.name
+    dataset_name = dataset.name
 
     # Start labeling.
     response = label_text.label_text(
-        dataset.name,
-        instruction.name,
-        annotation_spec_set.name
+        dataset_name,
+        instruction_name,
+        annotation_spec_set_name
     )
     out, _ = capsys.readouterr()
     assert 'Label_text operation name: ' in out
diff --git a/datalabeling/label_video_test.py b/datalabeling/label_video_test.py
index 606b28c1c9a..fd466b052c8 100644
--- a/datalabeling/label_video_test.py
+++ b/datalabeling/label_video_test.py
@@ -88,12 +88,15 @@ def instruction():
 # to be deleted before the annotation_spec_set can be deleted.
 @pytest.mark.slow
 def test_label_video(capsys, annotation_spec_set, instruction, dataset):
+    annotation_spec_set_name = annotation_spec_set.name
+    instruction_name = instruction.name
+    dataset_name = dataset.name
 
     # Start labeling.
     response = label_video.label_video(
-        dataset.name,
-        instruction.name,
-        annotation_spec_set.name
+        dataset_name,
+        instruction_name,
+        annotation_spec_set_name
     )
     out, _ = capsys.readouterr()
     assert 'Label_video operation name: ' in out