def test_variant_array_dim(): v = ua.Variant([1, 2, 3, 4, 5, 6], dimensions=[2, 3]) assert v.Value[1] == 2 v2 = variant_from_binary(ua.utils.Buffer(variant_to_binary(v))) assert _reshape(v.Value, (2, 3)) == v2.Value assert v.VariantType == v2.VariantType assert v.Dimensions == v2.Dimensions assert v2.Dimensions == [2, 3]
def test_flatten(): 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) assert [2, 3, 4] == dims assert l != l2 l3 = _reshape(l2, (2, 3, 4)) assert l == l3 l = [[[], [], []], [[], [], []]] l2 = flatten(l) dims = get_shape(l) assert dims == [2, 3, 0] l = [1, 2, 3, 4] l2 = flatten(l) dims = get_shape(l) assert dims == [4] assert l == l2