1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
|
# *****************************************************************************
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# See NOTICE file for details.
#
# *****************************************************************************
import pytest
import _jpype
import jpype
import logging
from os import path
import sys
import unittest
import platform
CLASSPATH = None
fast = False
def version(v):
return tuple([int(i) for i in v.split('.')])
def requirePythonAfter(required):
pversion = tuple([int(i) for i in platform.python_version_tuple()])
def g(func):
def f(self):
if pversion < required:
raise unittest.SkipTest("numpy required")
return func(self)
return f
return g
def requireInstrumentation(func):
def f(self):
import _jpype
if not hasattr(_jpype, "fault"):
raise unittest.SkipTest("instrumentation required")
rc = func(self)
_jpype.fault(None)
return rc
return f
def requireNumpy(func):
def f(self):
try:
import numpy
return func(self)
except ImportError:
pass
raise unittest.SkipTest("numpy required")
return f
class UseFunc(object):
def __init__(self, obj, func, attr):
self.obj = obj
self.func = func
self.attr = attr
self.orig = getattr(self.obj, self.attr)
def __enter__(self):
setattr(self.obj, self.attr, self.func)
def __exit__(self, exception_type, exception_value, traceback):
setattr(self.obj, self.attr, self.orig)
@pytest.mark.usefixtures("common_opts")
class JPypeTestCase(unittest.TestCase):
def setUp(self):
if not jpype.isJVMStarted():
try:
import faulthandler
faulthandler.enable()
faulthandler.disable()
except:
pass
root = path.dirname(path.abspath(path.dirname(__file__)))
jpype.addClassPath(path.join(root, 'classes'))
jvm_path = jpype.getDefaultJVMPath()
logger = logging.getLogger(__name__)
logger.info("Running testsuite using JVM %s" % jvm_path)
classpath_arg = "-Djava.class.path=%s"
args = ["-ea", "-Xmx256M", "-Xms16M"]
if self._checkjni:
args.append("-Xcheck:jni")
# TODO: enabling this check crashes the JVM with: FATAL ERROR in native method: Bad global or local ref passed to JNI
# "-Xcheck:jni",
if self._classpath:
from pathlib import Path
import warnings
# This needs to be relative to run location
jpype.addClassPath(Path(self._classpath).resolve())
warnings.warn("using jar instead of thunks")
if self._convertStrings:
import warnings
warnings.warn("using deprecated convertStrings")
if self._jacoco:
import warnings
args.append(
"-javaagent:lib/org.jacoco.agent-0.8.5-runtime.jar=destfile=build/coverage/jacoco.exec,includes=org.jpype.*")
warnings.warn("using JaCoCo")
import pathlib
jpype.addClassPath(pathlib.Path("lib/*").absolute())
jpype.addClassPath(pathlib.Path("test/jar/*").absolute())
classpath_arg %= jpype.getClassPath()
args.append(classpath_arg)
_jpype.enableStacktraces(True)
#JPypeTestCase.str_conversion = eval(os.getenv('JPYPE_STR_CONVERSION', 'True'))
jpype.startJVM(jvm_path, *args,
convertStrings=self._convertStrings)
self.jpype = jpype.JPackage('jpype')
if sys.version < '3':
self.assertCountEqual = self.assertItemsEqual
def tearDown(self):
pass
def assertElementsEqual(self, a, b):
self.assertEqual(len(a), len(b))
for i in range(len(a)):
self.assertEqual(a[i], b[i])
def assertElementsAlmostEqual(self, a, b):
self.assertEqual(len(a), len(b))
for i in range(len(a)):
self.assertAlmostEqual(a[i], b[i])
def useEqualityFunc(self, func):
return UseFunc(self, func, 'assertEqual')
if __name__ == '__main__':
unittest.main()
|