RustPython-test/extra_tests/snippets/builtin_memoryview.py at main · RustPython/RustPython-test · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import array

from testutils import assert_raises

obj = b"abcde"
a = memoryview(obj)
assert a.obj == obj

assert a[2:3] == b"c"

assert hash(obj) == hash(a)


class A(array.array): ...


class B(bytes): ...


class C: ...


memoryview(bytearray("abcde", encoding="utf-8"))
memoryview(array.array("i", [1, 2, 3]))
memoryview(A("b", [0]))
memoryview(B("abcde", encoding="utf-8"))

assert_raises(TypeError, lambda: memoryview([1, 2, 3]))
assert_raises(TypeError, lambda: memoryview((1, 2, 3)))
assert_raises(TypeError, lambda: memoryview({}))
assert_raises(TypeError, lambda: memoryview("string"))
assert_raises(TypeError, lambda: memoryview(C()))


def test_slice():
    b = b"123456789"
    m = memoryview(b)
    m2 = memoryview(b)
    assert m == m
    assert m == m2
    assert m.tobytes() == b"123456789"
    assert m == b
    assert m[::2].tobytes() == b"13579"
    assert m[::2] == b"13579"
    assert m[1::2].tobytes() == b"2468"
    assert m[::2][1:].tobytes() == b"3579"
    assert m[::2][1:-1].tobytes() == b"357"
    assert m[::2][::2].tobytes() == b"159"
    assert m[::2][1::2].tobytes() == b"37"
    assert m[::-1].tobytes() == b"987654321"
    assert m[::-2].tobytes() == b"97531"


test_slice()


def test_resizable():
    b = bytearray(b"123")
    b.append(4)
    m = memoryview(b)
    assert_raises(BufferError, lambda: b.append(5))
    m.release()
    b.append(6)
    m2 = memoryview(b)
    m4 = memoryview(m2)
    assert_raises(BufferError, lambda: b.append(5))
    m3 = memoryview(m2)
    assert_raises(BufferError, lambda: b.append(5))
    m2.release()
    assert_raises(BufferError, lambda: b.append(5))
    m3.release()
    m4.release()
    b.append(7)


test_resizable()


def test_delitem():
    a = b"abc"
    b = memoryview(a)
    assert_raises(TypeError, lambda: b.__delitem__())
    assert_raises(TypeError, lambda: b.__delitem__(0))
    assert_raises(TypeError, lambda: b.__delitem__(10))
    a = bytearray(b"abc")
    b = memoryview(a)
    assert_raises(TypeError, lambda: b.__delitem__())
    assert_raises(TypeError, lambda: b.__delitem__(1))
    assert_raises(TypeError, lambda: b.__delitem__(12))


test_delitem()