Make sorting benchmarks all sort integers

numpy · Apr 7, 2007 · a083ce5 · a083ce5
1 parent ada65fd
commit a083ce5
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/benchmarks/sorting.py b/benchmarks/sorting.py
@@ -5,21 +5,21 @@
 
 N = 10000
 b.title = 'Sorting %d elements' % N
-b['numarray'] = ('a=N.array(None,shape=%d);a.sort()'%N,'')
-b['numpy'] = ('a=N.empty(shape=%d);a.sort()'%N,'')
-b['Numeric'] = ('a=N.empty(shape=%d);N.sort(a)'%N,'')
+b['numarray'] = ('a=N.array(None,shape=%d,typecode="i");a.sort()'%N,'')
+b['numpy'] = ('a=N.empty(shape=%d, dtype="i");a.sort()'%N,'')
+b['Numeric'] = ('a=N.empty(shape=%d, typecode="i");N.sort(a)'%N,'')
 b.run()
 
 N1,N2 = 100,100
 b.title = 'Sorting (%d,%d) elements, last axis' % (N1,N2)
-b['numarray'] = ('a=N.array(None,shape=(%d,%d));a.sort()'%(N1,N2),'')
-b['numpy'] = ('a=N.empty(shape=(%d,%d));a.sort()'%(N1,N2),'')
-b['Numeric'] = ('a=N.empty(shape=(%d,%d));N.sort(a)'%(N1,N2),'')
+b['numarray'] = ('a=N.array(None,shape=(%d,%d),typecode="i");a.sort()'%(N1,N2),'')
+b['numpy'] = ('a=N.empty(shape=(%d,%d), dtype="i");a.sort()'%(N1,N2),'')
+b['Numeric'] = ('a=N.empty(shape=(%d,%d),typecode="i");N.sort(a)'%(N1,N2),'')
 b.run()
 
 N1,N2 = 100,100
 b.title = 'Sorting (%d,%d) elements, first axis' % (N1,N2)
-b['numarray'] = ('a=N.array(None,shape=(%d,%d));a.sort(0)'%(N1,N2),'')
-b['Numeric'] = ('a=N.empty(shape=(%d,%d));N.sort(a,0)'%(N1,N2),'')
-b['numpy'] = ('a=N.empty(shape=(%d,%d));N.sort(a,0)'%(N1,N2),'')
+b['numarray'] = ('a=N.array(None,shape=(%d,%d), typecode="i");a.sort(0)'%(N1,N2),'')
+b['numpy'] = ('a=N.empty(shape=(%d,%d),dtype="i");N.sort(a,0)'%(N1,N2),'')
+b['Numeric'] = ('a=N.empty(shape=(%d,%d),typecode="i");N.sort(a,0)'%(N1,N2),'')
 b.run()