Skip to content

Commit 95c8e39

Browse files
sbinetsbinet
committed
stdlib/array: add support for 'u' arrays
Signed-off-by: Sebastien Binet <binet@cern.ch>
1 parent 53252dd commit 95c8e39

File tree

3 files changed

+188
-67
lines changed

3 files changed

+188
-67
lines changed

‎stdlib/array/array.go‎

Lines changed: 69 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,11 @@ import (
1313
"github.com/go-python/gpython/py"
1414
)
1515

16+
// FIXME(sbinet): consider creating an "array handler" type for each of the typecodes
17+
// and make the handler a field of the "array" type.
18+
// or make "array" an interface ?
19+
20+
// array provides efficient manipulation of C-arrays (as Go slices).
1621
type array struct {
1722
descr byte // typecode of elements
1823
esize int // element size in bytes
@@ -197,12 +202,12 @@ func array_new(metatype *py.Type, args py.Tuple, kwargs py.StringDict) (py.Objec
197202
esize: descr2esize[descr[0]],
198203
}
199204

200-
if descr[0] == 'u' {
201-
// FIXME(sbinet)
202-
return nil, py.NotImplementedError
203-
}
204-
205205
switch descr[0] {
206+
case 'u':
207+
var data []rune
208+
arr.data = data
209+
arr.append = arr.appendRune
210+
arr.extend = arr.extendRune
206211
case 'b':
207212
var data []int8
208213
arr.data = data
@@ -300,14 +305,22 @@ func (arr *array) M__repr__() (py.Object, error) {
300305
o := new(strings.Builder)
301306
o.WriteString("array('" + string(arr.descr) + "'")
302307
if data := reflect.ValueOf(arr.data); arr.data != nil && data.Len() > 0 {
303-
o.WriteString(", [")
304-
for i := 0; i < data.Len(); i++ {
305-
if i > 0 {
306-
o.WriteString(", ")
308+
switch arr.descr {
309+
case 'u':
310+
o.WriteString(", '")
311+
o.WriteString(string(arr.data.([]rune)))
312+
o.WriteString("'")
313+
default:
314+
o.WriteString(", [")
315+
for i := 0; i < data.Len(); i++ {
316+
if i > 0 {
317+
o.WriteString(", ")
318+
}
319+
// FIXME(sbinet): we don't get exactly the same display wrt CPython for float32
320+
fmt.Fprintf(o, "%v", data.Index(i))
307321
}
308-
fmt.Fprintf(o, "%v", data.Index(i))
322+
o.WriteString("]")
309323
}
310-
o.WriteString("]")
311324
}
312325
o.WriteString(")")
313326
return py.String(o.String()), nil
@@ -344,8 +357,7 @@ func (arr *array) M__getitem__(k py.Object) (py.Object, error) {
344357
case 'B', 'H', 'I', 'L', 'Q':
345358
return py.Int(sli.Index(i).Uint()), nil
346359
case 'u':
347-
// FIXME(sbinet)
348-
return nil, py.NotImplementedError
360+
return py.String([]rune{rune(sli.Index(i).Int())}), nil
349361
case 'f', 'd':
350362
return py.Float(sli.Index(i).Float()), nil
351363
}
@@ -372,14 +384,23 @@ func (arr *array) M__setitem__(k, v py.Object) (py.Object, error) {
372384
}
373385
switch arr.descr {
374386
case 'b', 'h', 'i', 'l', 'q':
375-
vv := v.(py.Int)
387+
vv, ok := v.(py.Int)
388+
if !ok {
389+
return nil, py.ExceptionNewf(py.TypeError, "'%s' object cannot be interpreted as an integer", v.Type().Name)
390+
}
376391
sli.Index(i).SetInt(int64(vv))
377392
case 'B', 'H', 'I', 'L', 'Q':
378-
vv := v.(py.Int)
393+
vv, ok := v.(py.Int)
394+
if !ok {
395+
return nil, py.ExceptionNewf(py.TypeError, "'%s' object cannot be interpreted as an integer", v.Type().Name)
396+
}
379397
sli.Index(i).SetUint(uint64(vv))
380398
case 'u':
381-
// FIXME(sbinet)
382-
return nil, py.NotImplementedError
399+
vv, ok := v.(py.Int)
400+
if !ok {
401+
return nil, py.ExceptionNewf(py.TypeError, "array item must be unicode character")
402+
}
403+
sli.Index(i).SetInt(int64(vv))
383404
case 'f', 'd':
384405
var vv float64
385406
switch v := v.(type) {
@@ -401,6 +422,16 @@ func (arr *array) M__setitem__(k, v py.Object) (py.Object, error) {
401422
panic("impossible")
402423
}
403424

425+
func (arr *array) appendRune(v py.Object) (py.Object, error) {
426+
str, ok := v.(py.String)
427+
if !ok {
428+
return nil, py.ExceptionNewf(py.TypeError, "array item must be unicode character")
429+
}
430+
431+
arr.data = append(arr.data.([]rune), []rune(str)...)
432+
return py.None, nil
433+
}
434+
404435
func (arr *array) appendI8(v py.Object) (py.Object, error) {
405436
vv, err := asInt(v)
406437
if err != nil {
@@ -491,6 +522,27 @@ func (arr *array) appendF64(v py.Object) (py.Object, error) {
491522
return py.None, nil
492523
}
493524

525+
func (arr *array) extendRune(arg py.Object) (py.Object, error) {
526+
itr, err := py.Iter(arg)
527+
if err != nil {
528+
return nil, err
529+
}
530+
531+
nxt := itr.(py.I__next__)
532+
533+
for {
534+
o, err := nxt.M__next__()
535+
if err == py.StopIteration {
536+
break
537+
}
538+
_, err = arr.appendRune(o)
539+
if err != nil {
540+
return nil, err
541+
}
542+
}
543+
return py.None, nil
544+
}
545+
494546
func (arr *array) extendI8(arg py.Object) (py.Object, error) {
495547
itr, err := py.Iter(arg)
496548
if err != nil {

‎stdlib/array/testdata/test.py‎

Lines changed: 41 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -19,87 +19,105 @@ def assertEqual(x, y):
1919
print("")
2020
print("typecode '%s'" % (typ,))
2121
if typ == 'u':
22-
# FIXME(sbinet): implement
23-
print(" SKIP: NotImplemented")
24-
continue
25-
if typ in "bhilqfd":
22+
arr = array.array(typ, "?世界!")
23+
if typ in "bhilq":
2624
arr = array.array(typ, [-1, -2, -3, -4])
2725
if typ in "BHILQ":
2826
arr = array.array(typ, [+1, +2, +3, +4])
29-
print(" array: %s" % (repr(arr),))
27+
if typ in "fd":
28+
arr = array.array(typ, [-1.0, -2.0, -3.0, -4.0])
29+
print(" array: %s ## repr" % (repr(arr),))
30+
print(" array: %s ## str" % (str(arr),))
3031
print(" itemsize: %s" % (arr.itemsize,))
3132
print(" typecode: %s" % (arr.typecode,))
3233
print(" len: %s" % (len(arr),))
3334
print(" arr[0]: %s" % (arr[0],))
3435
print(" arr[-1]: %s" % (arr[-1],))
3536
try:
3637
arr[-10]
37-
print(" ERROR: expected an exception")
38+
print(" ERROR1: expected an exception")
3839
except:
3940
print(" caught an exception [ok]")
4041

4142
try:
4243
arr[10]
43-
print(" ERROR: expected an exception")
44+
print(" ERROR2: expected an exception")
4445
except:
4546
print(" caught an exception [ok]")
4647
arr[-2] = 33
48+
if typ in "fd":
49+
arr[-2] = 0.3
4750
print(" arr[-2]: %s" % (arr[-2],))
4851

4952
try:
5053
arr[-10] = 2
51-
print(" ERROR: expected an exception")
54+
print(" ERROR3: expected an exception")
5255
except:
5356
print(" caught an exception [ok]")
5457

5558
if typ in "bhilqfd":
5659
arr.extend([-5,-6])
5760
if typ in "BHILQ":
5861
arr.extend([5,6])
62+
if typ == 'u':
63+
arr.extend("he")
5964
print(" array: %s" % (repr(arr),))
6065
print(" len: %s" % (len(arr),))
6166

6267
if typ in "bhilqfd":
6368
arr.append(-7)
6469
if typ in "BHILQ":
6570
arr.append(7)
71+
if typ == 'u':
72+
arr.append("l")
6673
print(" array: %s" % (repr(arr),))
6774
print(" len: %s" % (len(arr),))
6875

6976
try:
7077
arr.append()
71-
print(" ERROR: expected an exception")
78+
print(" ERROR4: expected an exception")
7279
except:
7380
print(" caught an exception [ok]")
7481
try:
7582
arr.append([])
76-
print(" ERROR: expected an exception")
83+
print(" ERROR5: expected an exception")
7784
except:
7885
print(" caught an exception [ok]")
7986
try:
8087
arr.append(1, 2)
81-
print(" ERROR: expected an exception")
88+
print(" ERROR6: expected an exception")
8289
except:
8390
print(" caught an exception [ok]")
8491
try:
8592
arr.append(None)
86-
print(" ERROR: expected an exception")
93+
print(" ERROR7: expected an exception")
8794
except:
8895
print(" caught an exception [ok]")
8996

9097
try:
9198
arr.extend()
92-
print(" ERROR: expected an exception")
99+
print(" ERROR8: expected an exception")
93100
except:
94101
print(" caught an exception [ok]")
95102
try:
96103
arr.extend(None)
97-
print(" ERROR: expected an exception")
104+
print(" ERROR9: expected an exception")
98105
except:
99106
print(" caught an exception [ok]")
100107
try:
101108
arr.extend([1,None])
102-
print(" ERROR: expected an exception")
109+
print(" ERROR10: expected an exception")
110+
except:
111+
print(" caught an exception [ok]")
112+
try:
113+
arr.extend(1,None)
114+
print(" ERROR11: expected an exception")
115+
except:
116+
print(" caught an exception [ok]")
117+
118+
try:
119+
arr[0] = object()
120+
print(" ERROR12: expected an exception")
103121
except:
104122
print(" caught an exception [ok]")
105123
pass
@@ -108,55 +126,48 @@ def assertEqual(x, y):
108126
print("## testing array.array(...)")
109127
try:
110128
arr = array.array()
111-
print("ERROR: expected an exception")
129+
print("ERROR1: expected an exception")
112130
except:
113131
print("caught an exception [ok]")
114132

115133
try:
116134
arr = array.array(b"d")
117-
print("ERROR: expected an exception")
135+
print("ERROR2: expected an exception")
118136
except:
119137
print("caught an exception [ok]")
120138

121139
try:
122140
arr = array.array("?")
123-
print("ERROR: expected an exception")
141+
print("ERROR3: expected an exception")
124142
except:
125143
print("caught an exception [ok]")
126144

127145
try:
128146
arr = array.array("dd")
129-
print("ERROR: expected an exception")
147+
print("ERROR4: expected an exception")
130148
except:
131149
print("caught an exception [ok]")
132150

133151
try:
134152
arr = array.array("d", initializer=[1,2])
135-
print("ERROR: expected an exception")
153+
print("ERROR5: expected an exception")
136154
except:
137155
print("caught an exception [ok]")
138156

139157
try:
140158
arr = array.array("d", [1], [])
141-
print("ERROR: expected an exception")
159+
print("ERROR6: expected an exception")
142160
except:
143161
print("caught an exception [ok]")
144162

145163
try:
146164
arr = array.array("d", 1)
147-
print("ERROR: expected an exception")
165+
print("ERROR7: expected an exception")
148166
except:
149167
print("caught an exception [ok]")
150168

151169
try:
152170
arr = array.array("d", ["a","b"])
153-
print("ERROR: expected an exception")
154-
except:
155-
print("caught an exception [ok]")
156-
157-
try:
158-
## FIXME(sbinet): implement it at some point.
159-
arr = array.array("u")
160-
print("ERROR: expected an exception")
171+
print("ERROR8: expected an exception")
161172
except:
162173
print("caught an exception [ok]")

0 commit comments

Comments
 (0)