def test_VT_UNKNOWN_1(self):
a = _midlSAFEARRAY(POINTER(IUnknown))
t = _midlSAFEARRAY(POINTER(IUnknown))
self.failUnless(a is t)
def com_refcnt(o):
"Return the COM refcount of an interface pointer"
import gc
gc.collect()
gc.collect()
o.AddRef()
return o.Release()
from comtypes.typeinfo import CreateTypeLib, ICreateTypeLib
punk = CreateTypeLib("spam").QueryInterface(
IUnknown) # will never be saved to disk
# initial refcount
initial = com_refcnt(punk)
# This should increase the refcount by 1
sa = t.from_param([punk])
self.failUnlessEqual(initial + 1, com_refcnt(punk))
# Unpacking the array must not change the refcount, and must
# return an equal object.
self.failUnlessEqual((punk, ), sa[0])
self.failUnlessEqual(initial + 1, com_refcnt(punk))
del sa
self.failUnlessEqual(initial, com_refcnt(punk))
sa = t.from_param([None])
self.failUnlessEqual((POINTER(IUnknown)(), ), sa[0])
def test_UDT_ndarray(self):
np = get_numpy()
if np is None:
return
from comtypes.gen.TestComServerLib import MYCOLOR
t = _midlSAFEARRAY(MYCOLOR)
self.failUnless(t is _midlSAFEARRAY(MYCOLOR))
sa = t.from_param([MYCOLOR(0, 0, 0), MYCOLOR(1, 2, 3)])
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
# The conversion code allows numpy to choose the dtype of
# structured data. This dtype is structured under numpy 1.5, 1.7 and
# 1.8, and object in 1.6. Instead of assuming either of these, check
# the array contents based on the chosen type.
if arr.dtype is np.dtype(object):
data = [(x.red, x.green, x.blue) for x in arr]
else:
float_dtype = np.dtype('float64')
self.assertIs(arr.dtype[0], float_dtype)
self.assertIs(arr.dtype[1], float_dtype)
self.assertIs(arr.dtype[2], float_dtype)
data = [tuple(x) for x in arr]
self.failUnlessEqual(data, [(0.0, 0.0, 0.0), (1.0, 2.0, 3.0)])
def test_UDT_ndarray(self):
np = get_numpy()
if np is None:
return
from comtypes.gen.TestComServerLib import MYCOLOR
t = _midlSAFEARRAY(MYCOLOR)
self.failUnless(t is _midlSAFEARRAY(MYCOLOR))
sa = t.from_param([MYCOLOR(0, 0, 0), MYCOLOR(1, 2, 3)])
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
# The conversion code allows numpy to choose the dtype of
# structured data. This dtype is structured under numpy 1.5, 1.7 and
# 1.8, and object in 1.6. Instead of assuming either of these, check
# the array contents based on the chosen type.
if arr.dtype is np.dtype(object):
data = [(x.red, x.green, x.blue) for x in arr]
else:
float_dtype = np.dtype('float64')
self.assertIs(arr.dtype[0], float_dtype)
self.assertIs(arr.dtype[1], float_dtype)
self.assertIs(arr.dtype[2], float_dtype)
data = [tuple(x) for x in arr]
self.failUnlessEqual(data, [(0.0, 0.0, 0.0), (1.0, 2.0, 3.0)])
def test_VT_UNKNOWN_1(self):
a = _midlSAFEARRAY(POINTER(IUnknown))
t = _midlSAFEARRAY(POINTER(IUnknown))
self.assertTrue(a is t)
from comtypes.typeinfo import CreateTypeLib
# will never be saved to disk
punk = CreateTypeLib("spam").QueryInterface(IUnknown)
# initial refcount
initial = com_refcnt(punk)
# This should increase the refcount by 1
sa = t.from_param([punk])
self.assertEqual(initial + 1, com_refcnt(punk))
# Unpacking the array must not change the refcount, and must
# return an equal object.
self.assertEqual((punk, ), sa[0])
self.assertEqual(initial + 1, com_refcnt(punk))
del sa
self.assertEqual(initial, com_refcnt(punk))
sa = t.from_param([None])
self.assertEqual((POINTER(IUnknown)(), ), sa[0])
def test_VT_UNKNOWN_1(self):
a = _midlSAFEARRAY(POINTER(IUnknown))
t = _midlSAFEARRAY(POINTER(IUnknown))
self.failUnless(a is t)
def com_refcnt(o):
"Return the COM refcount of an interface pointer"
import gc; gc.collect(); gc.collect()
o.AddRef()
return o.Release()
from comtypes.typeinfo import CreateTypeLib, ICreateTypeLib
punk = CreateTypeLib("spam").QueryInterface(IUnknown) # will never be saved to disk
# initial refcount
initial = com_refcnt(punk)
# This should increase the refcount by 1
sa = t.from_param([punk])
self.failUnlessEqual(initial + 1, com_refcnt(punk))
# Unpacking the array must not change the refcount, and must
# return an equal object.
self.failUnlessEqual((punk,), sa[0])
self.failUnlessEqual(initial + 1, com_refcnt(punk))
del sa
self.failUnlessEqual(initial, com_refcnt(punk))
sa = t.from_param([None])
self.failUnlessEqual((POINTER(IUnknown)(),), sa[0])
def test_VT_UNKNOWN_multi_ndarray(self):
np = get_numpy()
if np is None:
return
a = _midlSAFEARRAY(POINTER(IUnknown))
t = _midlSAFEARRAY(POINTER(IUnknown))
self.failUnless(a is t)
def com_refcnt(o):
"Return the COM refcount of an interface pointer"
import gc
gc.collect()
gc.collect()
o.AddRef()
return o.Release()
from comtypes.typeinfo import CreateTypeLib, ICreateTypeLib
punk = CreateTypeLib("spam").QueryInterface(
IUnknown) # will never be saved to disk
# initial refcount
initial = com_refcnt(punk)
# This should increase the refcount by 4
sa = t.from_param((punk, ) * 4)
self.failUnlessEqual(initial + 4, com_refcnt(punk))
# Unpacking the array must not change the refcount, and must
# return an equal object. Creating an ndarray may change the
# refcount.
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(object), arr.dtype)
self.failUnless((arr == (punk, ) * 4).all())
self.failUnlessEqual(initial + 8, com_refcnt(punk))
del arr
self.failUnlessEqual(initial + 4, com_refcnt(punk))
del sa
self.failUnlessEqual(initial, com_refcnt(punk))
# This should increase the refcount by 2
sa = t.from_param((punk, None, punk, None))
self.failUnlessEqual(initial + 2, com_refcnt(punk))
null = POINTER(IUnknown)()
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(object), arr.dtype)
self.failUnless((arr == (punk, null, punk, null)).all())
del sa
del arr
self.failUnlessEqual(initial, com_refcnt(punk))
def test_VT_UNKNOWN_multi(self):
a = _midlSAFEARRAY(POINTER(IUnknown))
t = _midlSAFEARRAY(POINTER(IUnknown))
self.failUnless(a is t)
def com_refcnt(o):
"Return the COM refcount of an interface pointer"
import gc
gc.collect()
gc.collect()
o.AddRef()
return o.Release()
from comtypes.typeinfo import CreateTypeLib, ICreateTypeLib
punk = CreateTypeLib("spam").QueryInterface(
IUnknown) # will never be saved to disk
# initial refcount
initial = com_refcnt(punk)
# This should increase the refcount by 4
sa = t.from_param((punk, ) * 4)
self.failUnlessEqual(initial + 4, com_refcnt(punk))
# Unpacking the array must not change the refcount, and must
# return an equal object.
self.failUnlessEqual((punk, ) * 4, sa[0])
self.failUnlessEqual(initial + 4, com_refcnt(punk))
del sa
self.failUnlessEqual(initial, com_refcnt(punk))
# This should increase the refcount by 2
sa = t.from_param((punk, None, punk, None))
self.failUnlessEqual(initial + 2, com_refcnt(punk))
null = POINTER(IUnknown)()
self.failUnlessEqual((punk, null, punk, null), sa[0])
del sa
self.failUnlessEqual(initial, com_refcnt(punk))
# repeat same test, with 2 different com pointers
plib = CreateTypeLib("foo")
a, b = com_refcnt(plib), com_refcnt(punk)
sa = t.from_param([plib, punk, plib])
#### self.failUnlessEqual((plib, punk, plib), sa[0])
self.failUnlessEqual((a + 2, b + 1),
(com_refcnt(plib), com_refcnt(punk)))
del sa
self.failUnlessEqual((a, b), (com_refcnt(plib), com_refcnt(punk)))
def test_VT_UNKNOWN_multi_ndarray(self):
np = get_numpy()
if np is None:
return
a = _midlSAFEARRAY(POINTER(IUnknown))
t = _midlSAFEARRAY(POINTER(IUnknown))
self.failUnless(a is t)
def com_refcnt(o):
"Return the COM refcount of an interface pointer"
import gc; gc.collect(); gc.collect()
o.AddRef()
return o.Release()
from comtypes.typeinfo import CreateTypeLib, ICreateTypeLib
punk = CreateTypeLib("spam").QueryInterface(IUnknown) # will never be saved to disk
# initial refcount
initial = com_refcnt(punk)
# This should increase the refcount by 4
sa = t.from_param((punk,) * 4)
self.failUnlessEqual(initial + 4, com_refcnt(punk))
# Unpacking the array must not change the refcount, and must
# return an equal object. Creating an ndarray may change the
# refcount.
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(object), arr.dtype)
self.failUnless((arr == (punk,)*4).all())
self.failUnlessEqual(initial + 8, com_refcnt(punk))
del arr
self.failUnlessEqual(initial + 4, com_refcnt(punk))
del sa
self.failUnlessEqual(initial, com_refcnt(punk))
# This should increase the refcount by 2
sa = t.from_param((punk, None, punk, None))
self.failUnlessEqual(initial + 2, com_refcnt(punk))
null = POINTER(IUnknown)()
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(object), arr.dtype)
self.failUnless((arr == (punk, null, punk, null)).all())
del sa
del arr
self.failUnlessEqual(initial, com_refcnt(punk))
def test_equality(self):
a = _midlSAFEARRAY(c_long)
b = _midlSAFEARRAY(c_long)
self.failUnless(a is b)
c = _midlSAFEARRAY(BSTR)
d = _midlSAFEARRAY(BSTR)
self.failUnless(c is d)
self.failIfEqual(a, c)
# XXX remove:
self.failUnlessEqual((a._itemtype_, a._vartype_), (c_long, VT_I4))
self.failUnlessEqual((c._itemtype_, c._vartype_), (BSTR, VT_BSTR))
def test_equality(self):
a = _midlSAFEARRAY(c_long)
b = _midlSAFEARRAY(c_long)
self.assertTrue(a is b)
c = _midlSAFEARRAY(BSTR)
d = _midlSAFEARRAY(BSTR)
self.assertTrue(c is d)
self.assertNotEqual(a, c)
# XXX remove:
self.assertEqual((a._itemtype_, a._vartype_), (c_long, VT_I4))
self.assertEqual((c._itemtype_, c._vartype_), (BSTR, VT_BSTR))
def test_VT_UNKNOWN_multi(self):
a = _midlSAFEARRAY(POINTER(IUnknown))
t = _midlSAFEARRAY(POINTER(IUnknown))
self.failUnless(a is t)
def com_refcnt(o):
"Return the COM refcount of an interface pointer"
import gc; gc.collect(); gc.collect()
o.AddRef()
return o.Release()
from comtypes.typeinfo import CreateTypeLib, ICreateTypeLib
punk = CreateTypeLib("spam").QueryInterface(IUnknown) # will never be saved to disk
# initial refcount
initial = com_refcnt(punk)
# This should increase the refcount by 4
sa = t.from_param((punk,) * 4)
self.failUnlessEqual(initial + 4, com_refcnt(punk))
# Unpacking the array must not change the refcount, and must
# return an equal object.
self.failUnlessEqual((punk,)*4, sa[0])
self.failUnlessEqual(initial + 4, com_refcnt(punk))
del sa
self.failUnlessEqual(initial, com_refcnt(punk))
# This should increase the refcount by 2
sa = t.from_param((punk, None, punk, None))
self.failUnlessEqual(initial + 2, com_refcnt(punk))
null = POINTER(IUnknown)()
self.failUnlessEqual((punk, null, punk, null), sa[0])
del sa
self.failUnlessEqual(initial, com_refcnt(punk))
# repeat same test, with 2 different com pointers
plib = CreateTypeLib("foo")
a, b = com_refcnt(plib), com_refcnt(punk)
sa = t.from_param([plib, punk, plib])
#### self.failUnlessEqual((plib, punk, plib), sa[0])
self.failUnlessEqual((a+2, b+1), (com_refcnt(plib), com_refcnt(punk)))
del sa
self.failUnlessEqual((a, b), (com_refcnt(plib), com_refcnt(punk)))
def test_UDT(self):
from comtypes.gen.TestComServerLib import MYCOLOR
t = _midlSAFEARRAY(MYCOLOR)
self.failUnless(t is _midlSAFEARRAY(MYCOLOR))
sa = t.from_param([MYCOLOR(0, 0, 0), MYCOLOR(1, 2, 3)])
self.failUnlessEqual([(x.red, x.green, x.blue) for x in sa[0]],
[(0.0, 0.0, 0.0), (1.0, 2.0, 3.0)])
def doit():
t.from_param([MYCOLOR(0, 0, 0), MYCOLOR(1, 2, 3)])
bytes = find_memleak(doit)
self.failIf(bytes, "Leaks %d bytes" % bytes)
def test_UDT(self):
from comtypes.gen.TestComServerLib import MYCOLOR
t = _midlSAFEARRAY(MYCOLOR)
self.failUnless(t is _midlSAFEARRAY(MYCOLOR))
sa = t.from_param([MYCOLOR(0, 0, 0), MYCOLOR(1, 2, 3)])
self.failUnlessEqual([(x.red, x.green, x.blue) for x in sa[0]],
[(0.0, 0.0, 0.0), (1.0, 2.0, 3.0)])
def doit():
t.from_param([MYCOLOR(0, 0, 0), MYCOLOR(1, 2, 3)])
bytes = find_memleak(doit)
self.failIf(bytes, "Leaks %d bytes" % bytes)
def test_array(self):
t = _midlSAFEARRAY(c_double)
pat = pointer(t())
try:
import numpy
except ImportError:
pass # numpy not available
else:
pat[0] = numpy.zeros(32, dtype=numpy.float)
self.failUnlessEqual(tuple(pat[0][0]),
(0.0,) * 32)
data = ((1.0, 2.0, 3.0),
(4.0, 5.0, 6.0),
(7.0, 8.0, 9.0))
a = numpy.array(data,
dtype=numpy.double)
pat[0] = a
self.failUnlessEqual(pat[0][0],
data)
data = ((1.0, 2.0), (3.0, 4.0), (5.0, 6.0))
a = numpy.array(data,
dtype=numpy.double,
order="F")
pat[0] = a
self.failUnlessEqual(pat[0][0],
data)
def test_array(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(c_double)
pat = pointer(t())
pat[0] = np.zeros(32, dtype=np.float)
arr = get_array(pat[0])
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(np.double), arr.dtype)
self.failUnless((arr == (0.0, ) * 32).all())
data = ((1.0, 2.0, 3.0), (4.0, 5.0, 6.0), (7.0, 8.0, 9.0))
a = np.array(data, dtype=np.double)
pat[0] = a
arr = get_array(pat[0])
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(np.double), arr.dtype)
self.failUnless((arr == data).all())
data = ((1.0, 2.0), (3.0, 4.0), (5.0, 6.0))
a = np.array(data, dtype=np.double, order="F")
pat[0] = a
arr = get_array(pat[0])
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(np.double), arr.dtype)
self.failUnlessEqual(pat[0][0], data)
def test_equality(self):
a = _midlSAFEARRAY(c_long)
b = _midlSAFEARRAY(c_long)
self.failUnless(a is b)
c = _midlSAFEARRAY(BSTR)
d = _midlSAFEARRAY(BSTR)
self.failUnless(c is d)
self.failIfEqual(a, c)
# XXX remove:
self.failUnlessEqual((a._itemtype_, a._vartype_),
(c_long, VT_I4))
self.failUnlessEqual((c._itemtype_, c._vartype_),
(BSTR, VT_BSTR))
def test_VT_BSTR_leaks(self):
sb = _midlSAFEARRAY(BSTR)
def doit():
sb.from_param(["foo", "bar"])
bytes = find_memleak(doit)
self.failIf(bytes, "Leaks %d bytes" % bytes)
def test_VT_I4_leaks(self):
sa = _midlSAFEARRAY(c_long)
def doit():
sa.from_param([1, 2, 3, 4, 5, 6])
bytes = find_memleak(doit)
self.failIf(bytes, "Leaks %d bytes" % bytes)
def test_VT_BSTR_leaks(self):
sb = _midlSAFEARRAY(BSTR)
def doit():
sb.from_param(["foo", "bar"])
bytes = find_memleak(doit)
self.failIf(bytes, "Leaks %d bytes" % bytes)
def get_commit_message(self, parameters, logmessage):
commonurl = ""
commonroot = ""
bugid = ""
bstrarray = _midlSAFEARRAY(comtypes.BSTR)
pathlist = bstrarray.from_param(())
bugtr = self._get_bugtraq_object()
if bugtr is None:
return ""
try:
if self.supports_bugtraq2_interface():
(bugid, revPropNames, revPropValues,
newmessage) = bugtr.GetCommitMessage2(0, parameters,
commonurl, commonroot,
pathlist, logmessage,
bugid)
else:
newmessage = bugtr.GetCommitMessage(0, parameters, commonroot,
pathlist, logmessage)
except COMError:
qtlib.ErrorMsgBox(
_('Issue Tracker Plugin Error'),
_('Error getting commit message information from Issue Tracker plugin'
))
return ""
return newmessage
def test_array(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(c_double)
pat = pointer(t())
pat[0] = np.zeros(32, dtype=np.float)
arr = get_array(pat[0])
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(np.double), arr.dtype)
self.failUnless((arr == (0.0,) * 32).all())
data = ((1.0, 2.0, 3.0),
(4.0, 5.0, 6.0),
(7.0, 8.0, 9.0))
a = np.array(data, dtype=np.double)
pat[0] = a
arr = get_array(pat[0])
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(np.double), arr.dtype)
self.failUnless((arr == data).all())
data = ((1.0, 2.0), (3.0, 4.0), (5.0, 6.0))
a = np.array(data,
dtype=np.double,
order="F")
pat[0] = a
arr = get_array(pat[0])
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(np.double), arr.dtype)
self.failUnlessEqual(pat[0][0], data)
def test_array(self):
t = _midlSAFEARRAY(c_double)
pat = pointer(t())
try:
import numpy
except ImportError:
pass # numpy not available
else:
pat[0] = numpy.zeros(32, dtype=numpy.float)
self.failUnlessEqual(tuple(pat[0][0]),
(0.0,) * 32)
data = ((1.0, 2.0, 3.0),
(4.0, 5.0, 6.0),
(7.0, 8.0, 9.0))
a = numpy.array(data,
dtype=numpy.double)
pat[0] = a
self.failUnlessEqual(pat[0][0],
data)
data = ((1.0, 2.0), (3.0, 4.0), (5.0, 6.0))
a = numpy.array(data,
dtype=numpy.double,
order="F")
pat[0] = a
self.failUnlessEqual(pat[0][0],
data)
def test_VT_I4_leaks(self):
sa = _midlSAFEARRAY(c_long)
def doit():
sa.from_param([1, 2, 3, 4, 5, 6])
bytes = find_memleak(doit)
self.failIf(bytes, "Leaks %d bytes" % bytes)
def test_VT_VARIANT(self):
t = _midlSAFEARRAY(VARIANT)
now = datetime.datetime.now()
sa = t.from_param([11, "22", None, True, now, Decimal("3.14")])
self.failUnlessEqual(sa[0], (11, "22", None, True, now, Decimal("3.14")))
self.failUnlessEqual(SafeArrayGetVartype(sa), VT_VARIANT)
def test_VT_VARIANT(self):
t = _midlSAFEARRAY(VARIANT)
now = datetime.datetime.now()
sa = t.from_param([11, "22", None, True, now, Decimal("3.14")])
self.assertEqual(sa[0], (11, "22", None, True, now, Decimal("3.14")))
self.assertEqual(SafeArrayGetVartype(sa), VT_VARIANT)
def test_UDT_ndarray(self):
np = get_numpy()
if np is None:
return
from comtypes.gen.TestComServerLib import MYCOLOR
t = _midlSAFEARRAY(MYCOLOR)
self.failUnless(t is _midlSAFEARRAY(MYCOLOR))
sa = t.from_param([MYCOLOR(0, 0, 0), MYCOLOR(1, 2, 3)])
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(object), arr.dtype)
self.failUnlessEqual([(x.red, x.green, x.blue) for x in arr],
[(0.0, 0.0, 0.0), (1.0, 2.0, 3.0)])
def test_UDT_ndarray(self):
np = get_numpy()
if np is None:
return
from comtypes.gen.TestComServerLib import MYCOLOR
t = _midlSAFEARRAY(MYCOLOR)
self.failUnless(t is _midlSAFEARRAY(MYCOLOR))
sa = t.from_param([MYCOLOR(0, 0, 0), MYCOLOR(1, 2, 3)])
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(object), arr.dtype)
self.failUnlessEqual([(x.red, x.green, x.blue) for x in arr],
[(0.0, 0.0, 0.0), (1.0, 2.0, 3.0)])
def test_VT_I4(self):
t = _midlSAFEARRAY(c_long)
sa = t.from_param([11, 22, 33])
self.failUnlessEqual(sa[0], (11, 22, 33))
self.failUnlessEqual(SafeArrayGetVartype(sa), VT_I4)
# TypeError: len() of unsized object
self.assertRaises(TypeError, lambda: t.from_param(object()))
def test_VT_I4(self):
t = _midlSAFEARRAY(c_long)
sa = t.from_param([11, 22, 33])
self.failUnlessEqual(sa[0], (11, 22, 33))
self.failUnlessEqual(SafeArrayGetVartype(sa), VT_I4)
# TypeError: len() of unsized object
self.assertRaises(TypeError, lambda: t.from_param(object()))
def on_commit_finished(self, logmessage):
if not self.supports_bugtraq2_interface():
return ""
commonroot = ""
bstrarray = _midlSAFEARRAY(comtypes.BSTR)
pathlist = bstrarray.from_param(())
bugtr = self._get_bugtraq_object()
errormessage = bugtr.OnCommitFinished(0, commonroot, pathlist,
logmessage, 0)
return errormessage
def test_VT_BOOL_ndarray(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(VARIANT_BOOL)
sa = t.from_param([True, False, True, False])
arr = get_array(sa)
self.failUnlessEqual(np.dtype(np.bool_), arr.dtype)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnless((arr == (True, False, True, False)).all())
def test_VT_BOOL_ndarray(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(VARIANT_BOOL)
sa = t.from_param([True, False, True, False])
arr = get_array(sa)
self.failUnlessEqual(np.dtype(np.bool_), arr.dtype)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnless((arr == (True, False, True, False)).all())
class IBugTraqProvider2(IBugTraqProvider):
_iid_ = GUID("{C5C85E31-2F9B-4916-A7BA-8E27D481EE83}")
_methods_ = [
COMMETHOD(
[], HRESULT, "GetCommitMessage2",
(['in'], comtypes.c_long, "hParentWnd"),
(['in'], comtypes.BSTR, "parameters"),
(['in'], comtypes.BSTR, "commonURL"),
(['in'], comtypes.BSTR, "commonRoot"),
(['in'], _midlSAFEARRAY(comtypes.BSTR), "pathList"),
(['in'], comtypes.BSTR, "originalMessage"),
(['in'], comtypes.BSTR, "bugID"),
(['out'], POINTER(comtypes.BSTR), "bugIDOut"),
(['out'], POINTER(_midlSAFEARRAY(comtypes.BSTR)), "revPropNames"),
(['out'], POINTER(_midlSAFEARRAY(comtypes.BSTR)), "revPropValues"),
(['out', 'retval'], POINTER(comtypes.BSTR), "pRetVal")),
COMMETHOD([], HRESULT, "CheckCommit",
(['in'], comtypes.c_long, "hParentWnd"),
(['in'], comtypes.BSTR, "parameters"),
(['in'], comtypes.BSTR, "commonURL"),
(['in'], comtypes.BSTR, "commonRoot"),
(['in'], _midlSAFEARRAY(comtypes.BSTR), "pathList"),
(['in'], comtypes.BSTR, "commitMessage"),
(['out', 'retval'], POINTER(comtypes.BSTR), "pRetVal")),
COMMETHOD([], HRESULT, "OnCommitFinished",
(['in'], comtypes.c_long, "hParentWnd"),
(['in'], comtypes.BSTR, "commonRoot"),
(['in'], _midlSAFEARRAY(comtypes.BSTR), "pathList"),
(['in'], comtypes.BSTR, "logMessage"),
(['in'], comtypes.c_long, "revision"),
(['out', 'retval'], POINTER(comtypes.BSTR), "pRetVal")),
COMMETHOD([], HRESULT, "HasOptions",
(['out', 'retval'], POINTER(comtypes.c_int16), "pRetVal")),
COMMETHOD([], HRESULT, "ShowOptionsDialog",
(['in'], comtypes.c_long, "hParentWnd"),
(['in'], comtypes.BSTR, "parameters"),
(['out', 'retval'], POINTER(comtypes.BSTR), "pRetVal"))
]
def test_VT_BSTR_ndarray(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(BSTR)
sa = t.from_param(["a", "b", "c"])
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype('<U1'), arr.dtype)
self.failUnless((arr == ("a", "b", "c")).all())
self.failUnlessEqual(SafeArrayGetVartype(sa), VT_BSTR)
def test_VT_VARIANT_ndarray(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(VARIANT)
now = datetime.datetime.now()
sa = t.from_param([11, "22", None, True, now, Decimal("3.14")])
arr = get_array(sa)
self.failUnlessEqual(np.dtype(object), arr.dtype)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnless((arr == (11, "22", None, True, now, Decimal("3.14"))).all())
self.failUnlessEqual(SafeArrayGetVartype(sa), VT_VARIANT)
def test_VT_BSTR_ndarray(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(BSTR)
sa = t.from_param(["a", "b", "c"])
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype('<U1'), arr.dtype)
self.failUnless((arr == ("a", "b", "c")).all())
self.failUnlessEqual(SafeArrayGetVartype(sa), VT_BSTR)
def test_VT_I4_ndarray(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(c_long)
sa = t.from_param([11, 22, 33])
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(np.int), arr.dtype)
self.failUnless((arr == (11, 22, 33)).all())
self.failUnlessEqual(SafeArrayGetVartype(sa), VT_I4)
def test_VT_I4_ndarray(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(c_long)
sa = t.from_param([11, 22, 33])
arr = get_array(sa)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnlessEqual(np.dtype(np.int), arr.dtype)
self.failUnless((arr == (11, 22, 33)).all())
self.failUnlessEqual(SafeArrayGetVartype(sa), VT_I4)
def test_VT_VARIANT_ndarray(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(VARIANT)
now = datetime.datetime.now()
sa = t.from_param([11, "22", None, True, now, Decimal("3.14")])
arr = get_array(sa)
self.failUnlessEqual(np.dtype(object), arr.dtype)
self.failUnless(isinstance(arr, np.ndarray))
self.failUnless(
(arr == (11, "22", None, True, now, Decimal("3.14"))).all())
self.failUnlessEqual(SafeArrayGetVartype(sa), VT_VARIANT)
def test_VT_I4_ndarray(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(c_long)
inarr = np.array([11, 22, 33])
sa = t.from_param(inarr)
arr = get_array(sa)
self.assertTrue(isinstance(arr, np.ndarray))
self.assertEqual(np.dtype(np.int), arr.dtype)
self.assertTrue((arr == inarr).all())
self.assertEqual(SafeArrayGetVartype(sa), VT_I4)
def get_commit_message(self, parameters, logmessage):
commonurl = ""
commonroot = ""
bugid = ""
bstrarray = _midlSAFEARRAY(comtypes.BSTR)
pathlist = bstrarray.from_param(())
bugtr = self._get_bugtraq_object()
if self.supports_bugtraq2_interface():
(bugid, revPropNames, revPropValues, newmessage) = bugtr.GetCommitMessage2(
0, parameters, commonurl, commonroot, pathlist, logmessage, bugid)
else:
newmessage = bugtr.GetCommitMessage(
0, parameters, commonroot, pathlist, logmessage)
return newmessage
def test_VT_VARIANT_ndarray(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(VARIANT)
now = datetime.datetime.now()
inarr = np.array(
[11, "22", "33", 44.0, None, True, now,
Decimal("3.14")]).reshape(2, 4)
sa = t.from_param(inarr)
arr = get_array(sa)
self.assertEqual(np.dtype(object), arr.dtype)
self.assertTrue(isinstance(arr, np.ndarray))
self.assertTrue((arr == inarr).all())
self.assertEqual(SafeArrayGetVartype(sa), VT_VARIANT)
class IBugTraqProvider(IUnknown):
_iid_ = GUID("{298B927C-7220-423C-B7B4-6E241F00CD93}")
_methods_ = [
COMMETHOD([], HRESULT, "ValidateParameters",
(['in'], comtypes.c_long, "hParentWnd"),
(['in'], comtypes.BSTR, "parameters"),
(['out', 'retval'], POINTER(comtypes.c_int16), "pRetVal")),
COMMETHOD([], HRESULT, "GetLinkText",
(['in'], comtypes.c_long, "hParentWnd"),
(['in'], comtypes.BSTR, "parameters"),
(['out', 'retval'], POINTER(comtypes.BSTR), "pRetVal")),
COMMETHOD([], HRESULT, "GetCommitMessage",
(['in'], comtypes.c_long, "hParentWnd"),
(['in'], comtypes.BSTR, "parameters"),
(['in'], comtypes.BSTR, "commonRoot"),
(['in'], _midlSAFEARRAY(comtypes.BSTR), "pathList"),
(['in'], comtypes.BSTR, "originalMessage"),
(['out', 'retval'], POINTER(comtypes.BSTR), "pRetVal"))
]
def on_commit_finished(self, logmessage):
if not self.supports_bugtraq2_interface():
return ""
commonroot = ""
bstrarray = _midlSAFEARRAY(comtypes.BSTR)
pathlist = bstrarray.from_param(())
bugtr = self._get_bugtraq_object()
if bugtr is None:
return ""
try:
errormessage = bugtr.OnCommitFinished(0, commonroot, pathlist,
logmessage, 0)
except COMError:
qtlib.ErrorMsgBox(_('Issue Tracker Plugin Error'),
_('Error executing "commit finished" trigger'))
return ""
return errormessage
def on_commit_finished(self, logmessage):
if not self.supports_bugtraq2_interface():
return ""
commonroot = ""
bstrarray = _midlSAFEARRAY(comtypes.BSTR)
pathlist = bstrarray.from_param(())
bugtr = self._get_bugtraq_object()
if bugtr is None:
return ""
try:
errormessage = bugtr.OnCommitFinished(0, commonroot, pathlist,
logmessage, 0)
except COMError:
qtlib.ErrorMsgBox(_('Issue Tracker Plugin Error'),
_('Error executing "commit finished" trigger'))
return ""
return errormessage
def test_datetime64_ndarray(self):
np = get_numpy()
if np is None:
return
try:
np.datetime64
except AttributeError:
return
dates = np.array([
np.datetime64("2000-01-01T05:30:00", "s"),
np.datetime64("1800-01-01T05:30:00", "ms"),
np.datetime64("2014-03-07T00:12:56", "us"),
np.datetime64("2000-01-01T12:34:56", "ns"),
])
t = _midlSAFEARRAY(VARIANT)
sa = t.from_param(dates)
arr = get_array(sa).astype(dates.dtype)
self.failUnless((dates == arr).all())
def test_datetime64_ndarray(self):
np = get_numpy()
if np is None:
return
try:
np.datetime64
except AttributeError:
return
dates = np.array([
np.datetime64("2000-01-01T05:30:00", "s"),
np.datetime64("1800-01-01T05:30:00", "ms"),
np.datetime64("2014-03-07T00:12:56", "us"),
np.datetime64("2000-01-01T12:34:56", "ns"),
])
t = _midlSAFEARRAY(VARIANT)
sa = t.from_param(dates)
arr = get_array(sa).astype(dates.dtype)
self.failUnless((dates == arr).all())
def test_nested_contexts(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(BSTR)
sa = t.from_param(["a", "b", "c"])
first = sa[0]
with safearray_as_ndarray:
second = sa[0]
with safearray_as_ndarray:
third = sa[0]
fourth = sa[0]
fifth = sa[0]
self.failUnless(isinstance(first, tuple))
self.failUnless(isinstance(second, np.ndarray))
self.failUnless(isinstance(third, np.ndarray))
self.failUnless(isinstance(fourth, np.ndarray))
self.failUnless(isinstance(fifth, tuple))
def test_nested_contexts(self):
np = get_numpy()
if np is None:
return
t = _midlSAFEARRAY(BSTR)
sa = t.from_param(["a", "b", "c"])
first = sa[0]
with safearray_as_ndarray:
second = sa[0]
with safearray_as_ndarray:
third = sa[0]
fourth = sa[0]
fifth = sa[0]
self.failUnless(isinstance(first, tuple))
self.failUnless(isinstance(second, np.ndarray))
self.failUnless(isinstance(third, np.ndarray))
self.failUnless(isinstance(fourth, np.ndarray))
self.failUnless(isinstance(fifth, tuple))
def get_commit_message(self, parameters, logmessage):
commonurl = ""
commonroot = ""
bugid = ""
bstrarray = _midlSAFEARRAY(comtypes.BSTR)
pathlist = bstrarray.from_param(())
bugtr = self._get_bugtraq_object()
if bugtr is None:
return ""
try:
if self.supports_bugtraq2_interface():
(bugid, revPropNames, revPropValues, newmessage) = bugtr.GetCommitMessage2(
0, parameters, commonurl, commonroot, pathlist, logmessage, bugid)
else:
newmessage = bugtr.GetCommitMessage(
0, parameters, commonroot, pathlist, logmessage)
except COMError:
qtlib.ErrorMsgBox(_('Issue Tracker Plugin Error'),
_('Error getting commit message information from Issue Tracker plugin'))
return ""
return newmessage
def test_VT_BOOL(self):
t = _midlSAFEARRAY(VARIANT_BOOL)
sa = t.from_param([True, False, True, False])
self.failUnlessEqual(sa[0], (True, False, True, False))
def test_VT_BSTR(self):
t = _midlSAFEARRAY(BSTR)
sa = t.from_param(["a", "b", "c"])
self.failUnlessEqual(sa[0], ("a", "b", "c"))
self.failUnlessEqual(SafeArrayGetVartype(sa), VT_BSTR)
def test_VT_BOOL(self):
t = _midlSAFEARRAY(VARIANT_BOOL)
sa = t.from_param([True, False, True, False])
self.failUnlessEqual(sa[0], (True, False, True, False))
def test_VT_BSTR(self):
t = _midlSAFEARRAY(BSTR)
sa = t.from_param(["a" ,"b", "c"])
self.failUnlessEqual(sa[0], ("a", "b", "c"))
self.failUnlessEqual(SafeArrayGetVartype(sa), VT_BSTR)
