# RUN: env PYTHONPATH=%ir2vec_python_dir %python %s %S/../Inputs/input.ll %ir2vec_test_vocab_dir/dummy_3D_nonzero_opc_vocab.json | FileCheck %s

import sys
import ir2vec

ll_file = sys.argv[1]
vocab_path = sys.argv[2]

vocab = ir2vec.loadVocab(vocab_path)
tool = ir2vec.initEmbedding(
    filename=ll_file, mode=ir2vec.IR2VecKind.Symbolic, vocab=vocab
)

# Success case
emb = tool.getFuncEmb("add")
print(f"SUCCESS: {emb.tolist()}")
# CHECK: SUCCESS: [38.0, 40.0, 42.0]

# Error: Function not found
try:
    tool.getFuncEmb("nonexistent")
except ValueError:
    print("ERROR: Function not found")
# CHECK: ERROR: Function not found