Skip to content

Commit 595d4dc

Browse files
SK-2270: fix unit tests
1 parent 8a6a267 commit 595d4dc

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

tests/vault/controller/test__detect.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ def test_deidentify_file_txt_success(self, mock_open, mock_basename, mock_base64
159159
word_count=1, char_count=1, size_in_kb=1,
160160
duration_in_seconds=None, page_count=None,
161161
slide_count=None, entities=[], run_id="runid123",
162-
status="SUCCESS", errors=None)) as mock_parse:
162+
status="SUCCESS")) as mock_parse:
163163
result = self.detect.deidentify_file(req)
164164

165165
mock_validate.assert_called_once()
@@ -184,7 +184,6 @@ def test_deidentify_file_txt_success(self, mock_open, mock_basename, mock_base64
184184
self.assertIsNone(result.page_count)
185185
self.assertIsNone(result.slide_count)
186186
self.assertEqual(result.entities, [])
187-
self.assertEqual(result.errors, None)
188187

189188
@patch("skyflow.vault.controller._detect.validate_deidentify_file_request")
190189
@patch("skyflow.vault.controller._detect.base64")
@@ -222,7 +221,7 @@ def test_deidentify_file_audio_success(self, mock_base64, mock_validate):
222221
word_count=1, char_count=1, size_in_kb=1,
223222
duration_in_seconds=1, page_count=None,
224223
slide_count=None, entities=[], run_id="runid456",
225-
status="SUCCESS", errors=None)) as mock_parse:
224+
status="SUCCESS")) as mock_parse:
226225
result = self.detect.deidentify_file(req)
227226
mock_validate.assert_called_once()
228227
files_api.deidentify_audio.assert_called_once()
@@ -263,8 +262,7 @@ def test_get_detect_run_success(self, mock_validate):
263262
return_value=DeidentifyFileResponse(file="file", type="txt", extension="txt", word_count=1,
264263
char_count=1, size_in_kb=1, duration_in_seconds=None,
265264
page_count=None, slide_count=None, entities=[],
266-
run_id="runid789", status="SUCCESS",
267-
errors=None)) as mock_parse:
265+
run_id="runid789", status="SUCCESS")) as mock_parse:
268266
result = self.detect.get_detect_run(req)
269267
mock_validate.assert_called_once()
270268
files_api.get_run.assert_called_once()
@@ -679,7 +677,6 @@ def test_deidentify_file_using_file_path(self, mock_open, mock_basename, mock_ba
679677
entities=[],
680678
run_id="runid123",
681679
status="SUCCESS",
682-
errors=None
683680
)) as mock_parse:
684681

685682
result = self.detect.deidentify_file(req)
@@ -710,4 +707,3 @@ def test_deidentify_file_using_file_path(self, mock_open, mock_basename, mock_ba
710707
self.assertIsNone(result.page_count)
711708
self.assertIsNone(result.slide_count)
712709
self.assertEqual(result.entities, [])
713-
self.assertEqual(result.errors, None)

0 commit comments

Comments
 (0)