Reenable pytests (#8)

binsync · Jan 20, 2024 · a154a77 · a154a77
1 parent d9a00e9
commit a154a77
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 6 deletions.
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -31,7 +31,5 @@ jobs:
         pip install .
     - name: Test with pytest
       run: |
-        # TODO: add install of models to test when we go public
-        varbert --help
-        # TODO: uncomment when we go public
-        # pytest tests/tests.py
+        varbert --download-models
+        pytest tests/tests.py
diff --git a/tests/tests.py b/tests/tests.py
@@ -3,7 +3,7 @@
 from typing import Dict
 
 from varbert import VariableRenamingAPI
-from yodalib.data import Function, FunctionArgument, FunctionHeader, StackVariable
+from libbs.artifacts import Function, FunctionArgument, FunctionHeader, StackVariable
 
 
 def function_with_new_names(function: Function, new_names: Dict[str, str]):

diff --git a/varbert/__init__.py b/varbert/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "2.0.9"
+__version__ = "2.0.10"
 
 import importlib.resources
 import tarfile

diff --git a/varbert/api.py b/varbert/api.py
@@ -116,6 +116,7 @@ def predict_for_functions(func_addrs: Optional[List[int]] = None, decompiler: Op
         dec_interface = DecompilerInterface.discover(force_decompiler=decompiler)
         varbert = VariableRenamingAPI(decompiler_interface=dec_interface)
         func_addrs = func_addrs if func_addrs else dec_interface.functions
+        func_addrs = [dec_interface.art_lifer.lift_addr(addr) for addr in func_addrs]
 
         # grab real functions, which require decompilation, and predict names for them
         total_suggested_funcs = 0