Ejemplo n.º 1
0
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)
Ejemplo n.º 2
0
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)
Ejemplo n.º 3
0
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)
Ejemplo n.º 4
0
    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)
Ejemplo n.º 5
0
    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)
Ejemplo n.º 6
0
    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)
Ejemplo n.º 7
0
    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)
Ejemplo n.º 8
0
    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)
Ejemplo n.º 9
0
    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)
Ejemplo n.º 10
0
def get_ending(args):  
    if args.llvm:
        return ".bc"
    if args.olibs:
        return ".o"
    else:
        return pyc.find_shared_ending()
Ejemplo n.º 11
0
    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)
Ejemplo n.º 12
0
    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)
Ejemplo n.º 13
0
    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)
Ejemplo n.º 14
0
    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)
Ejemplo n.º 15
0
# -*- 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
Ejemplo n.º 16
0
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


Ejemplo n.º 17
0
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: