def test1(self):
x = InMemoryGridAttributeStorage(5, 4, 3)
i = (0, 1, 2, 3, 4)
j = (1, 3, 1, 3, 1)
k = (0, 2, 0, 2, 0)
x.set_values_in_store((i, j, k), ['a', 'b'],
[2.0 | units.kg, 1.0 | units.m])
b, a = x.get_values_in_store(None, ['b', 'a'])
print b.shape, a.shape
self.assertEquals(b[0][1][0], 1.0 | units.m)
self.assertEquals(b[0][0][0], 0.0 | units.m)
self.assertEquals(a[0][1][0], 2.0 | units.kg)
self.assertEquals(a[1][3][2], 2.0 | units.kg)
self.assertEquals(a[1][2][2], 0.0 | units.kg)
(b, ) = x.get_values_in_store(
(numpy.s_[0:4], numpy.s_[1:4], numpy.s_[:]), ['a'])
self.assertEquals(b[0][0][0], 2.0 | units.kg)
self.assertEquals(b[0][0][2], 0.0 | units.kg)
self.assertEquals(b[1][2][2], 2.0 | units.kg)
self.assertEquals(b[2][0][0], 2.0 | units.kg)
self.assertEquals(b[3][2][2], 2.0 | units.kg)
self.assertEquals(b.sum(), 8.0 | units.kg)
self.assertEquals(sorted(x.get_defined_attribute_names()), ["a", "b"])
def test3(self):
x = InMemoryGridAttributeStorage(5,4,3)
i = (0,1,2,3,4)
j = (1,3,1,3,1)
k = (0,2,0,2,0)
x.set_values_in_store(
(i,j,k),
['a','b'],
[2.0 | units.kg, 1.0 | units.m]
)
b, a = x.get_values_in_store((0,1,0), ['b','a'])
print b, a
self.assertEquals(b, 1.0 | units.m)
self.assertEquals(a, 2.0 | units.kg)
b, a = x.get_values_in_store((0,0,0), ['b','a'])
print b, a
self.assertEquals(b, 0.0 | units.m)
self.assertEquals(a, 0.0 | units.kg)
def test1(self):
x = InMemoryGridAttributeStorage(5,4,3)
i = (0,1,2,3,4)
j = (1,3,1,3,1)
k = (0,2,0,2,0)
x.set_values_in_store(
(i,j,k),
['a','b'],
[2.0 | units.kg, 1.0 | units.m]
)
b, a = x.get_values_in_store(None, ['b','a'])
print b.shape, a.shape
self.assertEquals(b[0][1][0], 1.0 | units.m)
self.assertEquals(b[0][0][0], 0.0 | units.m)
self.assertEquals(a[0][1][0], 2.0 | units.kg)
self.assertEquals(a[1][3][2], 2.0 | units.kg)
self.assertEquals(a[1][2][2], 0.0 | units.kg)
(b,) = x.get_values_in_store((numpy.s_[0:4], numpy.s_[1:4], numpy.s_[:]), ['a'])
self.assertEquals(b[0][0][0], 2.0 | units.kg)
self.assertEquals(b[0][0][2], 0.0 | units.kg)
self.assertEquals(b[1][2][2], 2.0 | units.kg)
self.assertEquals(b[2][0][0], 2.0 | units.kg)
self.assertEquals(b[3][2][2], 2.0 | units.kg)
self.assertEquals(b.sum(), 8.0 | units.kg)
self.assertEquals(sorted(x.get_defined_attribute_names()), ["a", "b"])
def test3(self):
x = InMemoryGridAttributeStorage(5, 4, 3)
i = (0, 1, 2, 3, 4)
j = (1, 3, 1, 3, 1)
k = (0, 2, 0, 2, 0)
x.set_values_in_store((i, j, k), ['a', 'b'],
[2.0 | units.kg, 1.0 | units.m])
b, a = x.get_values_in_store((0, 1, 0), ['b', 'a'])
print b, a
self.assertEquals(b, 1.0 | units.m)
self.assertEquals(a, 2.0 | units.kg)
b, a = x.get_values_in_store((0, 0, 0), ['b', 'a'])
print b, a
self.assertEquals(b, 0.0 | units.m)
self.assertEquals(a, 0.0 | units.kg)
def test2(self):
x = InMemoryGridAttributeStorage(5, 4, 3)
i = (0, 1, 2, 3, 4)
j = (1, 3, 1, 3, 1)
k = (0, 2, 0, 2, 0)
x.set_values_in_store((i, j, k), ['a', 'b'], [2.0, 1.0])
b, a = x.get_values_in_store(None, ['b', 'a'])
print(a)
self.assertEqual(b[0][1][0], 1.0)
self.assertEqual(b[0][0][0], 0.0)
self.assertEqual(a[0][1][0], 2.0)
self.assertEqual(a[1][3][2], 2.0)
self.assertEqual(a[1][2][2], 0.0)
(b, ) = x.get_values_in_store(
(numpy.s_[0:4], numpy.s_[1:4], numpy.s_[:]), ['a'])
self.assertEqual(b[0][0][0], 2.0)
self.assertEqual(b[0][0][2], 0.0)
self.assertEqual(b[1][2][2], 2.0)
self.assertEqual(b[2][0][0], 2.0)
self.assertEqual(b[3][2][2], 2.0)
self.assertEqual(b.sum(), 8.0)
self.assertEqual(sorted(x.get_defined_attribute_names()), ["a", "b"])
y = x.copy()
(b, ) = y.get_values_in_store(
(numpy.s_[0:4], numpy.s_[1:4], numpy.s_[:]), ['a'])
self.assertEqual(b[0][0][0], 2.0)
self.assertEqual(b[0][0][2], 0.0)
self.assertEqual(b[1][2][2], 2.0)
