|
1 |
| -import time |
| 1 | +# import time |
2 | 2 |
|
3 |
| -import pytest |
4 |
| -from packaging import version |
| 3 | +# import pytest |
| 4 | +# from packaging import version |
5 | 5 |
|
6 |
| -from arango.exceptions import ( |
7 |
| - PregelJobCreateError, |
8 |
| - PregelJobDeleteError, |
9 |
| - PregelJobGetError, |
10 |
| -) |
11 |
| -from tests.helpers import assert_raises, generate_string |
| 6 | +# from arango.exceptions import ( |
| 7 | +# PregelJobCreateError, |
| 8 | +# PregelJobDeleteError, |
| 9 | +# PregelJobGetError, |
| 10 | +# ) |
| 11 | +# from tests.helpers import assert_raises, generate_string |
12 | 12 |
|
13 | 13 |
|
14 |
| -def test_pregel_attributes(db, username): |
15 |
| - assert db.pregel.context in ["default", "async", "batch", "transaction"] |
16 |
| - assert db.pregel.username == username |
17 |
| - assert db.pregel.db_name == db.name |
18 |
| - assert repr(db.pregel) == f"<Pregel in {db.name}>" |
| 14 | +# def test_pregel_attributes(db, username): |
| 15 | +# assert db.pregel.context in ["default", "async", "batch", "transaction"] |
| 16 | +# assert db.pregel.username == username |
| 17 | +# assert db.pregel.db_name == db.name |
| 18 | +# assert repr(db.pregel) == f"<Pregel in {db.name}>" |
19 | 19 |
|
20 | 20 |
|
21 |
| -def test_pregel_management(db, db_version, graph, cluster): |
22 |
| - if cluster: |
23 |
| - pytest.skip("Not tested in a cluster setup") |
| 21 | +# def test_pregel_management(db, db_version, graph, cluster): |
| 22 | +# if cluster: |
| 23 | +# pytest.skip("Not tested in a cluster setup") |
24 | 24 |
|
25 |
| - # Test create pregel job |
26 |
| - job_id = db.pregel.create_job( |
27 |
| - graph.name, |
28 |
| - "pagerank", |
29 |
| - store=False, |
30 |
| - max_gss=100, |
31 |
| - thread_count=1, |
32 |
| - async_mode=False, |
33 |
| - result_field="result", |
34 |
| - algorithm_params={"threshold": 0.000001}, |
35 |
| - ) |
36 |
| - assert isinstance(job_id, int) |
| 25 | +# # Test create pregel job |
| 26 | +# job_id = db.pregel.create_job( |
| 27 | +# graph.name, |
| 28 | +# "pagerank", |
| 29 | +# store=False, |
| 30 | +# max_gss=100, |
| 31 | +# thread_count=1, |
| 32 | +# async_mode=False, |
| 33 | +# result_field="result", |
| 34 | +# algorithm_params={"threshold": 0.000001}, |
| 35 | +# ) |
| 36 | +# assert isinstance(job_id, int) |
37 | 37 |
|
38 |
| - # Test create pregel job with unsupported algorithm |
39 |
| - with assert_raises(PregelJobCreateError) as err: |
40 |
| - db.pregel.create_job(graph.name, "invalid") |
41 |
| - assert err.value.error_code in {4, 10, 1600} |
| 38 | +# # Test create pregel job with unsupported algorithm |
| 39 | +# with assert_raises(PregelJobCreateError) as err: |
| 40 | +# db.pregel.create_job(graph.name, "invalid") |
| 41 | +# assert err.value.error_code in {4, 10, 1600} |
42 | 42 |
|
43 |
| - # Test get existing pregel job |
44 |
| - job = db.pregel.job(job_id) |
45 |
| - assert isinstance(job["state"], str) |
46 |
| - assert isinstance(job["aggregators"], dict) |
47 |
| - assert isinstance(job["gss"], int) |
48 |
| - assert isinstance(job["received_count"], int) |
49 |
| - assert isinstance(job["send_count"], int) |
50 |
| - assert "total_runtime" in job |
| 43 | +# # Test get existing pregel job |
| 44 | +# job = db.pregel.job(job_id) |
| 45 | +# assert isinstance(job["state"], str) |
| 46 | +# assert isinstance(job["aggregators"], dict) |
| 47 | +# assert isinstance(job["gss"], int) |
| 48 | +# assert isinstance(job["received_count"], int) |
| 49 | +# assert isinstance(job["send_count"], int) |
| 50 | +# assert "total_runtime" in job |
51 | 51 |
|
52 |
| - # Test delete existing pregel job |
53 |
| - assert db.pregel.delete_job(job_id) is True |
54 |
| - time.sleep(0.2) |
55 |
| - if db_version < version.parse("3.11.0"): |
56 |
| - with assert_raises(PregelJobGetError) as err: |
57 |
| - db.pregel.job(job_id) |
58 |
| - assert err.value.error_code in {4, 10, 1600} |
59 |
| - else: |
60 |
| - job = db.pregel.job(job_id) |
61 |
| - assert job["state"] == "canceled" |
| 52 | +# # Test delete existing pregel job |
| 53 | +# assert db.pregel.delete_job(job_id) is True |
| 54 | +# time.sleep(0.2) |
| 55 | +# if db_version < version.parse("3.11.0"): |
| 56 | +# with assert_raises(PregelJobGetError) as err: |
| 57 | +# db.pregel.job(job_id) |
| 58 | +# assert err.value.error_code in {4, 10, 1600} |
| 59 | +# else: |
| 60 | +# job = db.pregel.job(job_id) |
| 61 | +# assert job["state"] == "canceled" |
62 | 62 |
|
63 |
| - # Test delete missing pregel job |
64 |
| - with assert_raises(PregelJobDeleteError) as err: |
65 |
| - db.pregel.delete_job(generate_string()) |
66 |
| - assert err.value.error_code in {4, 10, 1600} |
| 63 | +# # Test delete missing pregel job |
| 64 | +# with assert_raises(PregelJobDeleteError) as err: |
| 65 | +# db.pregel.delete_job(generate_string()) |
| 66 | +# assert err.value.error_code in {4, 10, 1600} |
0 commit comments