Skip to content

Commit 7c16633

Browse files
committed
Add arrow debug endpoint
1 parent b3a2b20 commit 7c16633

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

graphdatascience/system/system_endpoints.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,12 @@ def sysInfo(self) -> "Series[Any]":
1515

1616
return self._query_runner.run_query(query).squeeze() # type: ignore
1717

18+
def arrow(self) -> "Series[Any]":
19+
self._namespace += ".arrow"
20+
query = f"CALL {self._namespace}()"
21+
22+
return self._query_runner.run_query(query).squeeze() # type: ignore
23+
1824

1925
class DirectSystemEndpoints(CallerBase):
2026
@client_only_endpoint("gds")

graphdatascience/tests/integration/test_system_ops.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,12 @@ def test_sysInfo(gds: GraphDataScience) -> None:
2929

3030
assert "gdsVersion" in (list(result["key"]))
3131

32+
@pytest.mark.skip_on_aura
33+
def test_sysInfo(gds: GraphDataScience) -> None:
34+
result = gds.debug.arrow()
35+
36+
assert "listenAddress" in (list(result["key"]))
37+
3238

3339
@pytest.mark.enterprise
3440
def test_is_licensed(gds: GraphDataScience) -> None:

0 commit comments

Comments
 (0)