本文整理汇总了Python中distarray.local.localarray.LocalArray.global_limits方法的典型用法代码示例。如果您正苦于以下问题:Python LocalArray.global_limits方法的具体用法?Python LocalArray.global_limits怎么用?Python LocalArray.global_limits使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类distarray.local.localarray.LocalArray
的用法示例。
在下文中一共展示了LocalArray.global_limits方法的3个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_global_limits_cyclic
# 需要导入模块: from distarray.local.localarray import LocalArray [as 别名]
# 或者: from distarray.local.localarray.LocalArray import global_limits [as 别名]
def test_global_limits_cyclic(self):
"""Find the boundaries of a cyclic distribution"""
d = Distribution.from_shape((16, 16), dist=('c', 'n'), comm=self.comm)
a = LocalArray(d)
answers = [(0, 12), (1, 13), (2, 14), (3, 15)]
limits = a.global_limits(0)
self.assertEqual(limits, answers[a.comm_rank])
answers = 4 * [(0, 15)]
limits = a.global_limits(1)
self.assertEqual(limits, answers[a.comm_rank])
示例2: test_global_limits_block
# 需要导入模块: from distarray.local.localarray import LocalArray [as 别名]
# 或者: from distarray.local.localarray.LocalArray import global_limits [as 别名]
def test_global_limits_block(self):
"""Find the boundaries of a block distribution"""
d = Distribution.from_shape((16, 16), dist=('b', 'n'), comm=self.comm)
a = LocalArray(d)
answers = [(0, 3), (4, 7), (8, 11), (12, 15)]
limits = a.global_limits(0)
self.assertEqual(limits, answers[a.comm_rank])
answers = 4 * [(0, 15)]
limits = a.global_limits(1)
self.assertEqual(limits, answers[a.comm_rank])
示例3: test_bad_global_limits
# 需要导入模块: from distarray.local.localarray import LocalArray [as 别名]
# 或者: from distarray.local.localarray.LocalArray import global_limits [as 别名]
def test_bad_global_limits(self):
""" Test that invalid global_limits fails as expected. """
d = Distribution.from_shape(comm=self.comm, shape=(4, 4))
a = LocalArray(d)
with self.assertRaises(InvalidDimensionError):
a.global_limits(-1)