def test_variant_array_dim(self):
v = ua.Variant([1, 2, 3, 4, 5, 6], dimensions = [2, 3])
self.assertEqual(v.Value[1], 2)
v2 = ua.Variant.from_binary(ua.utils.Buffer(v.to_binary()))
self.assertEqual(reshape(v.Value, (2,3)), v2.Value)
self.assertEqual(v.VariantType, v2.VariantType)
self.assertEqual(v.Dimensions, v2.Dimensions)
self.assertEqual(v2.Dimensions, [2, 3])
def test_flatten(self):
l = [[[1.0, 1.0, 1.0, 1.0], [1.0, 1.0, 1.0, 1.0], [1.0, 1.0, 1.0, 1.0]],[[1.0, 1.0, 1.0, 1.0], [1.0, 1.0, 1.0, 1.0], [1.0, 1.0, 1.0, 1.0]]]
l2 = flatten(l)
dims = get_shape(l)
self.assertEqual(dims, [2, 3, 4])
self.assertNotEqual(l, l2)
l3 = reshape(l2, (2,3,4))
self.assertEqual(l, l3)
l = [[[], [], []], [[], [], []]]
l2 = flatten(l)
dims = get_shape(l)
self.assertEqual(dims, [2, 3, 0])
l = [1, 2, 3, 4]
l2 = flatten(l)
dims = get_shape(l)
self.assertEqual(dims, [4])
self.assertEqual(l, l2)
