aniva/Pantograph
aniva
/
Pantograph
1
1
Fork 0
Code Issues 13 Pull Requests 4 Packages Projects Releases Wiki Activity

added more type hint and options; moved test_server_init_del below test_version

This commit is contained in:
Qi Liu 2025-01-14 11:09:38 +08:00
parent f4cc21cc96
commit ac36fd6cd0
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
pantograph/server.py
View File

@ -82,15 +82,16 @@ class Server:
@classmethod @classmethod
async def create(cls, async def create(cls,
imports=["Init"], imports: List[str]=["Init"],
project_path=None, project_path: Optional[str]=None,
lean_path=None, lean_path: Optional[str]=None,
# Options for executing the REPL. # Options for executing the REPL.
# Set `{ "automaticMode" : False }` to handle resumption by yourself. # Set `{ "automaticMode" : False }` to handle resumption by yourself.
options={}, options: Dict[str, Any]={},
core_options=DEFAULT_CORE_OPTIONS, core_options: List[str]=DEFAULT_CORE_OPTIONS,
timeout=120, timeout: int=120,
maxread=1000000) -> 'Server': maxread: int=1000000,
start:bool=True) -> 'Server':
""" """
timeout: Amount of time to wait for execution timeout: Amount of time to wait for execution
maxread: Maximum number of characters to read (especially important for large proofs and catalogs) maxread: Maximum number of characters to read (especially important for large proofs and catalogs)
@ -108,7 +109,8 @@ class Server:
if project_path and not lean_path: if project_path and not lean_path:
lean_path = await get_lean_path_async(project_path) lean_path = await get_lean_path_async(project_path)
self.lean_path = lean_path self.lean_path = lean_path
await self.restart_async() if start:
await self.restart_async()
return self return self
def __enter__(self) -> "Server": def __enter__(self) -> "Server":
@ -119,7 +121,7 @@ class Server:
def __del__(self): def __del__(self):
self._close() self._close()
def _close(self): def _close(self):
if self.proc is not None: if self.proc is not None:
try: try:
@ -419,6 +421,9 @@ def get_version():
class TestServer(unittest.TestCase): class TestServer(unittest.TestCase):
def test_version(self):
self.assertEqual(get_version(), "0.2.23")
def test_server_init_del(self): def test_server_init_del(self):
import warnings import warnings
@ -434,9 +439,6 @@ class TestServer(unittest.TestCase):
t = server.expr_type("forall (n m: Nat), n + m = m + n") t = server.expr_type("forall (n m: Nat), n + m = m + n")
del server del server
def test_version(self):
self.assertEqual(get_version(), "0.2.23")
def test_expr_type(self): def test_expr_type(self):
server = Server() server = Server()
t = server.expr_type("forall (n m: Nat), n + m = m + n") t = server.expr_type("forall (n m: Nat), n + m = m + n")