Skip to content

Commit

Permalink
Merge branch 'main' into remove-extra-group
Browse files Browse the repository at this point in the history
  • Loading branch information
fivegrant authored Sep 11, 2023
2 parents 7bc4d34 + 09a6498 commit fdac1ef
Showing 1 changed file with 8 additions and 10 deletions.
18 changes: 8 additions & 10 deletions tests/test_e2e.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ def test_pdf_extraction(context_dir, http_mock, client, worker, gen_tds_artifact
#### ASSERT ####
assert results.get("status") == "queued"
assert status_response.status_code == 200
assert status_response.json().get("status") == "finished"
assert status_response.json().get("status") == "finished", f"The RQ job failed.\n{job.latest_result().exc_string}"


@pytest.mark.parametrize("resource", params["pdf_to_text"])
Expand Down Expand Up @@ -89,7 +89,7 @@ def test_pdf_to_text(context_dir, http_mock, client, worker, gen_tds_artifact, f
#### ASSERT ####
assert results.get("status") == "queued"
assert status_response.status_code == 200
assert status_response.json().get("status") == "finished"
assert status_response.json().get("status") == "finished", f"The RQ job failed.\n{job.latest_result().exc_string}"


@pytest.mark.parametrize("resource", params["code_to_amr"])
Expand Down Expand Up @@ -132,13 +132,12 @@ def test_code_to_amr(context_dir, http_mock, client, worker, gen_tds_artifact, f
status_response = client.get(f"/status/{job_id}")

job = Job.fetch(job_id, connection=worker.connection)
print(job)
amr_instance = AMR(job.result["amr"])
if job.result is not None: amr_instance = AMR(job.result["amr"])

#### ASSERT ####
assert results.get("status") == "queued"
assert status_response.status_code == 200
assert status_response.json().get("status") == "finished"
assert status_response.json().get("status") == "finished", f"The RQ job failed.\n{job.latest_result().exc_string}"

assert (
amr_instance.is_valid()
Expand Down Expand Up @@ -184,12 +183,12 @@ def test_equations_to_amr(context_dir, http_mock, client, worker, file_storage):
status_response = client.get(f"/status/{job_id}")

job = Job.fetch(job_id, connection=worker.connection)
amr_instance = AMR(job.result["amr"])
if job.result is not None: amr_instance = AMR(job.result["amr"])

#### ASSERT ####
assert results.get("status") == "queued"
assert status_response.status_code == 200
assert status_response.json().get("status") == "finished"
assert status_response.json().get("status") == "finished", f"The RQ job failed.\n{job.latest_result().exc_string}"

assert (
amr_instance.is_valid()
Expand Down Expand Up @@ -249,8 +248,7 @@ def test_profile_dataset(context_dir, http_mock, client, worker, gen_tds_artifac
#### ASSERT ####
assert results.get("status") == "queued"
assert status_response.status_code == 200
assert status_response.json().get("status") == "finished"

assert status_response.json().get("status") == "finished", f"The RQ job failed.\n{job.latest_result().exc_string}"

@pytest.mark.parametrize("resource", params["profile_model"])
def test_profile_model(context_dir, http_mock, client, worker, gen_tds_artifact, file_storage, resource):
Expand Down Expand Up @@ -320,4 +318,4 @@ def test_profile_model(context_dir, http_mock, client, worker, gen_tds_artifact,
#### ASSERT ####
assert results.get("status") == "queued"
assert status_response.status_code == 200
assert status_response.json().get("status") == "finished"
assert status_response.json().get("status") == "finished", f"The RQ job failed.\n{job.latest_result().exc_string}"

0 comments on commit fdac1ef

Please sign in to comment.