def test_mismatch_errors(self):
ranks = getMPISizeWorld()
r = rRectangle(2 * ranks - 1, 2, l0=1., d0=ranks)
s = sRectangle(2, 2 * ranks, 2, l0=2., d0=ranks)
with self.assertRaises(RuntimeError): #length mismatch
self.calculateVariance(r, s)
r = rBrick(2 * ranks - 1, 2, 2, l1=2., d0=ranks)
s = sBrick(2, 2 * ranks, 2, 2, l1=1., d0=ranks)
with self.assertRaises(RuntimeError): #length mismatch
self.calculateVariance(r, s)
r = rRectangle(2 * ranks - 1, 2, l0=1., l1=2., d0=ranks)
s = sBrick(2, 2 * ranks, 2, 2, l0=2., l1=1., d0=ranks)
with self.assertRaises(RuntimeError): #dimensionality mismatch
self.calculateVariance(r, s)
if getMPISizeWorld() > 1:
r = rRectangle(2 * ranks - 1, 2, l0=1., d0=ranks)
s = sRectangle(2, 2, 2 * ranks, l0=2., d1=ranks)
with self.assertRaises(RuntimeError): #subdivision mismatch
self.calculateVariance(r, s)
def test_mismatch_errors(self):
ranks = getMPISizeWorld()
r = rRectangle(2 * ranks - 1, 2, l0=1.0, d0=ranks)
s = sRectangle(2, 2 * ranks, 2, l0=2.0, d0=ranks)
with self.assertRaises(RuntimeError): # length mismatch
self.calculateVariance(r, s)
r = rBrick(2 * ranks - 1, 2, 2, l1=2.0, d0=ranks)
s = sBrick(2, 2 * ranks, 2, 2, l1=1.0, d0=ranks)
with self.assertRaises(RuntimeError): # length mismatch
self.calculateVariance(r, s)
r = rRectangle(2 * ranks - 1, 2, l0=1.0, l1=2.0, d0=ranks)
s = sBrick(2, 2 * ranks, 2, 2, l0=2.0, l1=1.0, d0=ranks)
with self.assertRaises(RuntimeError): # dimensionality mismatch
self.calculateVariance(r, s)
if getMPISizeWorld() > 1:
r = rRectangle(2 * ranks - 1, 2, l0=1.0, d0=ranks)
s = sRectangle(2, 2, 2 * ranks, l0=2.0, d1=ranks)
with self.assertRaises(RuntimeError): # subdivision mismatch
self.calculateVariance(r, s)
def test_MultiDimSplit_Rectangle(self):
ranks = getMPISizeWorld()
squares = [4, 9, 16, 25, 36] #could do more, unlikely to test though
subdivs = [0, 0]
loop = 0
if ranks not in squares:
if ranks % 3 == 0:
subdivs[0] = 3
subdivs[1] = ranks // 3
loop = 2
elif ranks % 2 == 0:
subdivs[0] = 2
subdivs[1] = ranks // 2
loop = 1
else:
raise unittest.SkipTest("inappropriate number of ranks")
else:
subdivs = [squares.index(ranks) + 2] * 2
loop = 2
for order in range(2, 11):
divs = subdivs[:]
for i in range(loop):
r = rRectangle(2 * divs[0] * order - 1,
2 * divs[1] * order - 1,
d0=divs[0],
d1=divs[1])
s = sRectangle(order,
divs[0] * 2,
divs[1] * 2,
d0=divs[0],
d1=divs[1])
d = self.calculateVariance(s, r)
res = Lsup(d)
self.assertLess(
res, 1e-10, "".join([
"Rectangles of order %d with " % order,
"subdivisons %s " % divs,
"failed to interpolate correctly",
", variance of %g" % res
]))
divs.append(divs.pop(0))
def test_MultiDimSplit_Rectangle(self):
ranks = getMPISizeWorld()
squares = [4, 9, 16, 25, 36] # could do more, unlikely to test though
subdivs = [0, 0]
loop = 0
if ranks not in squares:
if ranks % 3 == 0:
subdivs[0] = 3
subdivs[1] = ranks // 3
loop = 2
elif ranks % 2 == 0:
subdivs[0] = 2
subdivs[1] = ranks // 2
loop = 1
else:
raise unittest.SkipTest("inappropriate number of ranks")
else:
subdivs = [squares.index(ranks) + 2] * 2
loop = 2
for order in range(2, 11):
divs = subdivs[:]
for i in range(loop):
r = rRectangle(2 * divs[0] * order - 1, 2 * divs[1] * order - 1, d0=divs[0], d1=divs[1])
s = sRectangle(order, divs[0] * 2, divs[1] * 2, d0=divs[0], d1=divs[1])
d = self.calculateVariance(s, r)
res = Lsup(d)
self.assertLess(
res,
1e-10,
"".join(
[
"Rectangles of order %d with " % order,
"subdivisons %s " % divs,
"failed to interpolate correctly",
", variance of %g" % res,
]
),
)
divs.append(divs.pop(0))
def createDomains(self, pointsPerDim):
"""
Creates a pair of domains with the previously supplied information and
number of points.
:param pointsPerDim: the number of data points (not elements) in each
dimension
:type pointsPerDim: ``tuple`` of ``int``
"""
splitDim = pointsPerDim.index(max(pointsPerDim))
divs = [1] * self.dim
divs[splitDim] = self.ranks
speckleyElements = []
ripleyElements = []
for i in range(self.dim):
points = pointsPerDim[i]
if i == splitDim:
ripleyElements.append(points + self.ranks -
(points % self.ranks) - 1)
else:
ripleyElements.append(points)
speck = ripleyElements[i] / self.order % self.ranks
if i == splitDim:
if speck != 0:
speck = self.ranks - speck
speck += ripleyElements[i] / self.order
if speck < 2:
speck = 2
speckleyElements.append(speck)
self.speckleyShape = tuple(speckleyElements)
self.ripleyShape = tuple(ripleyElements)
if self.dim == 2:
l0, l1 = self.lengths
ex, ey = speckleyElements
self.speckley = sRectangle(self.order,
ex,
ey,
d0=divs[0],
d1=divs[1],
l0=l0,
l1=l1,
diracPoints=self.diracPoints,
diracTags=self.diracTags)
ex, ey = ripleyElements
self.ripley = rRectangle(ex,
ey,
d0=divs[0],
d1=divs[1],
l0=l0,
l1=l1,
diracPoints=self.diracPoints,
diracTags=self.diracTags)
else:
l0, l1, l2 = self.lengths
ex, ey, ez = speckleyElements
self.speckley = sBrick(self.order,
ex,
ey,
ez,
d0=divs[0],
d1=divs[1],
d2=divs[2],
l0=l0,
l1=l1,
l2=l2,
diracPoints=self.diracPoints,
diracTags=self.diracTags)
ex, ey, ez = ripleyElements
self.ripley = rBrick(ex,
ey,
ez,
d0=divs[0],
d1=divs[1],
d2=divs[2],
l0=l0,
l1=l1,
l2=l2,
diracPoints=self.diracPoints,
diracTags=self.diracTags)
def createDomains(self, pointsPerDim):
"""
Creates a pair of domains with the previously supplied information and
number of points.
:param pointsPerDim: the number of data points (not elements) in each
dimension
:type pointsPerDim: ``tuple`` of ``int``
"""
splitDim = pointsPerDim.index(max(pointsPerDim))
divs = [1] * self.dim
divs[splitDim] = self.ranks
speckleyElements = []
ripleyElements = []
for i in range(self.dim):
points = pointsPerDim[i]
if i == splitDim:
ripleyElements.append(points + self.ranks - (points % self.ranks) - 1)
else:
ripleyElements.append(points)
speck = ripleyElements[i] / self.order % self.ranks
if i == splitDim:
if speck != 0:
speck = self.ranks - speck
speck += ripleyElements[i] / self.order
if speck < 2:
speck = 2
speckleyElements.append(speck)
self.speckleyShape = tuple(speckleyElements)
self.ripleyShape = tuple(ripleyElements)
if self.dim == 2:
l0, l1 = self.lengths
ex, ey = speckleyElements
self.speckley = sRectangle(
self.order,
ex,
ey,
d0=divs[0],
d1=divs[1],
l0=l0,
l1=l1,
diracPoints=self.diracPoints,
diracTags=self.diracTags,
)
ex, ey = ripleyElements
self.ripley = rRectangle(
ex, ey, d0=divs[0], d1=divs[1], l0=l0, l1=l1, diracPoints=self.diracPoints, diracTags=self.diracTags
)
else:
l0, l1, l2 = self.lengths
ex, ey, ez = speckleyElements
self.speckley = sBrick(
self.order,
ex,
ey,
ez,
d0=divs[0],
d1=divs[1],
d2=divs[2],
l0=l0,
l1=l1,
l2=l2,
diracPoints=self.diracPoints,
diracTags=self.diracTags,
)
ex, ey, ez = ripleyElements
self.ripley = rBrick(
ex,
ey,
ez,
d0=divs[0],
d1=divs[1],
d2=divs[2],
l0=l0,
l1=l1,
l2=l2,
diracPoints=self.diracPoints,
diracTags=self.diracTags,
)
