def test_pycc():
modulename = os.path.join(base_path, "compile_with_pycc")
cdll_modulename = modulename + find_shared_ending()
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
main(args=[modulename + ".py"])
lib = CDLL(cdll_modulename)
try:
lib.mult.argtypes = [c_double, c_double]
lib.mult.restype = c_double
lib.multf.argtypes = [c_float, c_float]
lib.multf.restype = c_float
res = lib.mult(123, 321)
print("lib.mult(123, 321) = %f", res)
assert res == 123 * 321
res = lib.multf(987, 321)
print("lib.multf(987, 321) = %f" % res)
assert res == 987 * 321
finally:
del lib
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
tmpdir = tempfile.gettempdir()
print("tmpdir: %s" % tmpdir)
out_modulename = os.path.join(tmpdir, "compiled_with_pycc") + find_shared_ending()
main(args=["--python", "-o", out_modulename, modulename + ".py"])
sys.path.append(tmpdir)
try:
import compiled_with_pycc as lib
try:
res = lib.mult(123, 321)
print("lib.mult(123, 321) = %f" % res)
assert res == 123 * 321
res = lib.multf(987, 321)
print("lib.multf(987, 321) = %f" % res)
assert res == 987 * 321
finally:
del lib
finally:
if os.path.exists(out_modulename):
os.unlink(out_modulename)
def test_pycc():
modulename = os.path.join(base_path, 'compile_with_pycc')
cdll_modulename = modulename + find_shared_ending()
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
main(args=[modulename + '.py'])
lib = CDLL(cdll_modulename)
try:
lib.mult.argtypes = [c_double, c_double]
lib.mult.restype = c_double
lib.multf.argtypes = [c_float, c_float]
lib.multf.restype = c_float
res = lib.mult(123, 321)
print('lib.mult(123, 321) = %f', res)
assert res == 123 * 321
res = lib.multf(987, 321)
print('lib.multf(987, 321) = %f' % res)
assert res == 987 * 321
finally:
del lib
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
tmpdir = tempfile.gettempdir()
print('tmpdir: %s' % tmpdir)
out_modulename = (os.path.join(tmpdir, 'compiled_with_pycc')
+ find_shared_ending())
main(args=['--python', '-o', out_modulename, modulename + '.py'])
sys.path.append(tmpdir)
try:
import compiled_with_pycc as lib
try:
res = lib.mult(123, 321)
print('lib.mult(123, 321) = %f' % res)
assert res == 123 * 321
res = lib.multf(987, 321)
print('lib.multf(987, 321) = %f' % res)
assert res == 987 * 321
finally:
del lib
finally:
if os.path.exists(out_modulename):
os.unlink(out_modulename)
def test_pycc():
modulename = os.path.join(os.path.dirname(__file__), 'compile_with_pycc')
cdll_modulename = modulename + find_shared_ending()
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
pycc.main(args=[modulename + '.py'])
lib = CDLL(cdll_modulename)
try:
lib.mult.argtypes = [c_double, c_double]
lib.mult.restype = c_double
lib.multf.argtypes = [c_float, c_float]
lib.multf.restype = c_float
res = lib.mult(123, 321)
print 'lib.mult(123, 321) =', res
assert res == 123 * 321
res = lib.multf(987, 321)
print 'lib.multf(987, 321) =', res
assert res == 987 * 321
finally:
del lib
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
out_modulename = (os.path.join(os.path.dirname(__file__),
'compiled_with_pycc')
+ find_shared_ending())
pycc.main(args=['--python', '-o', out_modulename, modulename + '.py'])
try:
import numba.tests.compiled_with_pycc as lib
try:
res = lib.mult(123, 321)
print 'lib.mult(123, 321) =', res
assert res == 123 * 321
res = lib.multf(987, 321)
print 'lib.multf(987, 321) =', res
assert res == 987 * 321
finally:
del lib
finally:
if os.path.exists(out_modulename):
os.unlink(out_modulename)
def test_pycc_pymodule(self):
unset_macosx_deployment_target()
modulename = os.path.join(base_path, 'compile_with_pycc')
tmpdir = tempfile.gettempdir()
print('tmpdir: %s' % tmpdir)
out_modulename = (os.path.join(tmpdir, 'compiled_with_pycc') +
find_shared_ending())
main(args=['--python', '-o', out_modulename, modulename + '.py'])
sys.path.append(tmpdir)
try:
import compiled_with_pycc as lib
try:
res = lib.mult(123, 321)
print('lib.mult(123, 321) = %f' % res)
assert res == 123 * 321
res = lib.multf(987, 321)
print('lib.multf(987, 321) = %f' % res)
assert res == 987 * 321
finally:
del lib
finally:
if os.path.exists(out_modulename):
os.unlink(out_modulename)
def test_pycc_ctypes_lib(self):
modulename = os.path.join(base_path, "compile_with_pycc")
cdll_modulename = modulename + find_shared_ending()
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
main(args=[modulename + ".py"])
lib = CDLL(cdll_modulename)
try:
lib.mult.argtypes = [POINTER(c_double), c_double, c_double]
lib.mult.restype = c_int
lib.multf.argtypes = [POINTER(c_float), c_float, c_float]
lib.multf.restype = c_int
res = c_double()
lib.mult(byref(res), 123, 321)
print("lib.mult(123, 321) = %f" % res.value)
self.assertEqual(res.value, 123 * 321)
res = c_float()
lib.multf(byref(res), 987, 321)
print("lib.multf(987, 321) = %f" % res.value)
self.assertEqual(res.value, 987 * 321)
finally:
del lib
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
def test_pycc_ctypes_lib(self):
modulename = os.path.join(base_path, 'compile_with_pycc')
cdll_modulename = modulename + find_shared_ending()
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
main(args=[modulename + '.py'])
lib = CDLL(cdll_modulename)
try:
lib.mult.argtypes = [POINTER(c_double), c_double, c_double]
lib.mult.restype = c_int
lib.multf.argtypes = [POINTER(c_float), c_float, c_float]
lib.multf.restype = c_int
res = c_double()
lib.mult(byref(res), 123, 321)
print('lib.mult(123, 321) = %f' % res.value)
self.assertEqual(res.value, 123 * 321)
res = c_float()
lib.multf(byref(res), 987, 321)
print('lib.multf(987, 321) = %f' % res.value)
self.assertEqual(res.value, 987 * 321)
finally:
del lib
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
def test_pycc_pymodule(self):
unset_macosx_deployment_target()
modulename = os.path.join(base_path, 'compile_with_pycc')
tmpdir = tempfile.gettempdir()
print('tmpdir: %s' % tmpdir)
out_modulename = (os.path.join(tmpdir, 'compiled_with_pycc')
+ find_shared_ending())
main(args=['--python', '-o', out_modulename, modulename + '.py'])
sys.path.append(tmpdir)
try:
import compiled_with_pycc as lib
try:
res = lib.mult(123, 321)
print('lib.mult(123, 321) = %f' % res)
assert res == 123 * 321
res = lib.multf(987, 321)
print('lib.multf(987, 321) = %f' % res)
assert res == 987 * 321
finally:
del lib
finally:
if os.path.exists(out_modulename):
os.unlink(out_modulename)
def test_pycc_ctypes_lib(self):
unset_macosx_deployment_target()
modulename = os.path.join(base_path, 'compile_with_pycc')
cdll_modulename = modulename + find_shared_ending()
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
main(args=[modulename + '.py'])
lib = CDLL(cdll_modulename)
try:
lib.mult.argtypes = [POINTER(c_double), c_void_p, c_double,
c_double]
lib.mult.restype = c_int
lib.multf.argtypes = [POINTER(c_float), c_void_p, c_float, c_float]
lib.multf.restype = c_int
res = c_double()
lib.mult(byref(res), None, 123, 321)
print('lib.mult(123, 321) = %f' % res.value)
self.assertEqual(res.value, 123 * 321)
res = c_float()
lib.multf(byref(res), None, 987, 321)
print('lib.multf(987, 321) = %f' % res.value)
self.assertEqual(res.value, 987 * 321)
finally:
del lib
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
def test_pycc_ctypes_lib(self):
"""
Test creating a C shared library object using pycc.
"""
unset_macosx_deployment_target()
source = os.path.join(base_path, 'compile_with_pycc.py')
cdll_modulename = 'test_dll_legacy' + find_shared_ending()
cdll_path = os.path.join(self.tmpdir, cdll_modulename)
if os.path.exists(cdll_path):
os.unlink(cdll_path)
main(args=['--debug', '-o', cdll_path, source])
lib = CDLL(cdll_path)
lib.mult.argtypes = [
POINTER(c_double), c_void_p, c_void_p, c_double, c_double
]
lib.mult.restype = c_int
lib.multf.argtypes = [
POINTER(c_float), c_void_p, c_void_p, c_float, c_float
]
lib.multf.restype = c_int
res = c_double()
lib.mult(byref(res), None, None, 123, 321)
self.assertEqual(res.value, 123 * 321)
res = c_float()
lib.multf(byref(res), None, None, 987, 321)
self.assertEqual(res.value, 987 * 321)
def get_ending(args):
if args.llvm:
return ".bc"
if args.olibs:
return ".o"
else:
return pyc.find_shared_ending()
def test_pycc_pymodule(self):
"""
Test creating a CPython extension module using pycc.
"""
unset_macosx_deployment_target()
modulename = os.path.join(base_path, 'compile_with_pycc')
tmpdir = tempfile.gettempdir()
out_modulename = (os.path.join(tmpdir, 'compiled_with_pycc')
+ find_shared_ending())
def _cleanup():
if os.path.exists(out_modulename):
os.unlink(out_modulename)
_cleanup()
self.addCleanup(_cleanup)
main(args=['--python', '-o', out_modulename, modulename + '.py'])
sys.path.append(tmpdir)
try:
import compiled_with_pycc as lib
try:
res = lib.mult(123, 321)
assert res == 123 * 321
res = lib.multf(987, 321)
assert res == 987 * 321
finally:
del lib
finally:
sys.path.remove(tmpdir)
def test_pycc_ctypes_lib(self):
"""
Test creating a C shared library object using pycc.
"""
unset_macosx_deployment_target()
modulename = os.path.join(base_path, 'compile_with_pycc')
cdll_modulename = modulename + find_shared_ending()
def _cleanup():
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
_cleanup()
self.addCleanup(_cleanup)
main(args=[modulename + '.py'])
lib = CDLL(cdll_modulename)
lib.mult.argtypes = [POINTER(c_double), c_void_p, c_void_p,
c_double, c_double]
lib.mult.restype = c_int
lib.multf.argtypes = [POINTER(c_float), c_void_p, c_void_p,
c_float, c_float]
lib.multf.restype = c_int
res = c_double()
lib.mult(byref(res), None, None, 123, 321)
self.assertEqual(res.value, 123 * 321)
res = c_float()
lib.multf(byref(res), None, None, 987, 321)
self.assertEqual(res.value, 987 * 321)
def test_pycc_ctypes_lib(self):
"""
Test creating a C shared library object using pycc.
"""
unset_macosx_deployment_target()
source = os.path.join(base_path, 'compile_with_pycc.py')
cdll_modulename = 'test_dll_legacy' + find_shared_ending()
cdll_path = os.path.join(self.tmpdir, cdll_modulename)
if os.path.exists(cdll_path):
os.unlink(cdll_path)
main(args=['--debug', '-o', cdll_path, source])
lib = CDLL(cdll_path)
lib.mult.argtypes = [POINTER(c_double), c_void_p, c_void_p,
c_double, c_double]
lib.mult.restype = c_int
lib.multf.argtypes = [POINTER(c_float), c_void_p, c_void_p,
c_float, c_float]
lib.multf.restype = c_int
res = c_double()
lib.mult(byref(res), None, None, 123, 321)
self.assertEqual(res.value, 123 * 321)
res = c_float()
lib.multf(byref(res), None, None, 987, 321)
self.assertEqual(res.value, 987 * 321)
def test_pycc_pymodule(self):
modulename = os.path.join(base_path, "compile_with_pycc")
tmpdir = tempfile.gettempdir()
print("tmpdir: %s" % tmpdir)
out_modulename = os.path.join(tmpdir, "compiled_with_pycc") + find_shared_ending()
main(args=["--python", "-o", out_modulename, modulename + ".py"])
sys.path.append(tmpdir)
try:
import compiled_with_pycc as lib
try:
res = lib.mult(123, 321)
print("lib.mult(123, 321) = %f" % res)
assert res == 123 * 321
res = lib.multf(987, 321)
print("lib.multf(987, 321) = %f" % res)
assert res == 987 * 321
finally:
del lib
finally:
if os.path.exists(out_modulename):
os.unlink(out_modulename)
# -*- coding: utf-8 -*-
from __future__ import print_function
import os
import tempfile
import sys
from ctypes import *
from numba.pycc import find_shared_ending, main
is_windows = sys.platform.startswith('win32')
if is_windows:
raise OSError('Example does not work on Windows platforms yet.')
base_path = os.path.dirname(os.path.abspath(__file__))
modulename = os.path.join(base_path, 'compile_with_pycc')
cdll_modulename = modulename + find_shared_ending()
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
# Compile python module to library
main(args=[modulename + '.py'])
lib = CDLL(cdll_modulename)
# Load library with ctypes and call mult function
try:
lib.mult.argtypes = [POINTER(c_double), c_double, c_double]
lib.mult.restype = c_int
lib.multf.argtypes = [POINTER(c_float), c_float, c_float]
lib.multf.restype = c_int
import os from ctypes import * import subprocess from numba.pycc import find_shared_ending from numba.pycc import pycc modulename = os.path.join(os.path.dirname(__file__), 'compile_with_pycc') pycc.main(args=[modulename + '.py']) lib = CDLL(modulename + find_shared_ending()) lib.mult.argtypes = [c_double, c_double] lib.mult.restype = c_double lib.multf.argtypes = [c_float, c_float] lib.multf.restype = c_float res = lib.mult(123, 321) print 'lib.mult(123, 321) =', res assert res == 123 * 321 res = lib.multf(987, 321) print 'lib.multf(987, 321) =', res assert res == 987 * 321
import os
from ctypes import *
from numba.pycc import find_shared_ending
from numba.pycc import pycc
modulename = os.path.join(os.path.dirname(__file__), 'compile_with_pycc')
cdll_modulename = modulename + find_shared_ending()
if os.path.exists(cdll_modulename):
os.unlink(cdll_modulename)
pycc.main(args=[modulename + '.py'])
lib = CDLL(cdll_modulename)
try:
lib.mult.argtypes = [c_double, c_double]
lib.mult.restype = c_double
lib.multf.argtypes = [c_float, c_float]
lib.multf.restype = c_float
res = lib.mult(123, 321)
print 'lib.mult(123, 321) =', res
assert res == 123 * 321
res = lib.multf(987, 321)
print 'lib.multf(987, 321) =', res
assert res == 987 * 321
finally: