Skip to content

Commit fcb0650

Browse files
authored
Merge pull request scottransom#7 from astrogreen/master
Updated tests to work with Python 3.4 and current version of numpy.
2 parents 4710ddf + a458ff9 commit fcb0650

File tree

1 file changed

+29
-27
lines changed

1 file changed

+29
-27
lines changed

test/test_slalib.py

+29-27
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from __future__ import unicode_literals
2+
13
from pyslalib import slalib as S
24
import numpy as N
35
import numpy.testing as T
@@ -216,13 +218,13 @@ def testcc62s(self):
216218

217219
def testcd2tf(self):
218220
s, ihmsf = S.sla_cd2tf(4, -0.987654321)
219-
T.assert_equal(s, '-', 'sla_cd2tf, s')
221+
T.assert_equal(s, b'-', 'sla_cd2tf, s')
220222
T.assert_equal(ihmsf[0], 23, 'sla_cd2tf, (1)')
221223
T.assert_equal(ihmsf[1], 42, 'sla_cd2tf, (2)')
222224
T.assert_equal(ihmsf[2], 13, 'sla_cd2tf, (3)')
223225
T.assert_almost_equal(ihmsf[3], 3333, -3, 'sla_cd2tf, (4)')
224226
s, ihmsf = S.sla_dd2tf(4, -0.987654321)
225-
T.assert_equal(s, '-', 'sla_dd2tf, s')
227+
T.assert_equal(s, b'-', 'sla_dd2tf, s')
226228
T.assert_equal(ihmsf[0], 23, 'sla_dd2tf, (1)')
227229
T.assert_equal(ihmsf[1], 42, 'sla_dd2tf, (2)')
228230
T.assert_equal(ihmsf[2], 13, 'sla_dd2tf, (3)')
@@ -235,27 +237,27 @@ def testcldj(self):
235237

236238
def testcr2af(self):
237239
s, idmsf = S.sla_cr2af(4, 2.345)
238-
T.assert_equal(s, '+', 'sla_cr2af, s')
240+
T.assert_equal(s, b'+', 'sla_cr2af, s')
239241
T.assert_equal(idmsf[0], 134, 'sla_cr2af, (1)')
240242
T.assert_equal(idmsf[1], 21, 'sla_cr2af, (2)')
241243
T.assert_equal(idmsf[2], 30, 'sla_cr2af, (3)')
242244
T.assert_almost_equal(idmsf[3], 9706, -3, 'sla_cr2af, (4)')
243245
s, idmsf = S.sla_dr2af(4, 2.345)
244-
T.assert_equal(s, '+', 'sla_dr2af, s')
246+
T.assert_equal(s, b'+', 'sla_dr2af, s')
245247
T.assert_equal(idmsf[0], 134, 'sla_dr2af, (1)')
246248
T.assert_equal(idmsf[1], 21, 'sla_dr2af, (2)')
247249
T.assert_equal(idmsf[2], 30, 'sla_dr2af, (3)')
248250
T.assert_equal(idmsf[3], 9706, 'sla_dr2af, (4)')
249251

250252
def testcr2tf(self):
251253
s, ihmsf = S.sla_cr2tf(4, -3.01234)
252-
T.assert_equal(s, '-', 'sla_cr2tf, s')
254+
T.assert_equal(s, b'-', 'sla_cr2tf, s')
253255
T.assert_equal(ihmsf[0], 11, 'sla_cr2tf, (1)')
254256
T.assert_equal(ihmsf[1], 30, 'sla_cr2tf, (2)')
255257
T.assert_equal(ihmsf[2], 22, 'sla_cr2tf, (3)')
256258
T.assert_almost_equal(ihmsf[3], 6484, -3, 'sla_cr2tf, (4)')
257259
s, ihmsf = S.sla_dr2tf(4, -3.01234)
258-
T.assert_equal(s, '-', 'sla_dr2tf, s')
260+
T.assert_equal(s, b'-', 'sla_dr2tf, s')
259261
T.assert_equal(ihmsf[0], 11, 'sla_dr2tf, (1)')
260262
T.assert_equal(ihmsf[1], 30, 'sla_dr2tf, (2)')
261263
T.assert_equal(ihmsf[2], 22, 'sla_dr2tf, (3)')
@@ -493,13 +495,13 @@ def testfitxy(self):
493495
45, 32.5, 8.6, -17,
494496
15.3, 10, 121.7, -3.8])
495497
# The reshaping is necessary due to C/Fortran order
496-
xye = N.reshape(xye, (2,8), order='Fortran')
498+
xye = N.reshape(xye, (2,8), order='F')
497499
xym = N.asarray([-23.41, 12.12, 32.03, 15.34,
498500
10.93, -23.72, -3.01, -16.10,
499501
44.90, -32.46, 8.55, 17.02,
500502
15.31, -10.07, 120.92, 3.81])
501503
# The reshaping is necessary due to C/Fortran order
502-
xym = N.reshape(xym, (2,8), order='Fortran')
504+
xym = N.reshape(xym, (2,8), order='F')
503505
coeffs, j = S.sla_fitxy(4, xye, xym)
504506
ans1 = N.asarray([-7.938263381515947e-3, 1.004640925187200,
505507
3.976948048238268e-4, -2.501031681585021e-2,
@@ -522,7 +524,7 @@ def testfitxy(self):
522524
15.348618307280820, 10.07063070741086835,
523525
121.5833272936291482, -3.788442308260240])
524526
# And another reshaping...
525-
ans3 = N.reshape(ans3, (2,8), order='Fortran')
527+
ans3 = N.reshape(ans3, (2,8), order='F')
526528
T.assert_array_almost_equal(xyp, ans3, 12, 'sla_fitxy, xyp')
527529
T.assert_almost_equal(xrms ,0.1087247110488075, 13, 'sla_pxy, xrms')
528530
T.assert_almost_equal(yrms, 0.03224481175794666, 13, 'sla_pxy, yrms')
@@ -721,29 +723,29 @@ def testkbj(self):
721723
k = '?'
722724
e = 1950
723725
k, j = S.sla_kbj(-1, e)
724-
T.assert_equal(k, ' ', 'sla_kbj, jb1')
726+
T.assert_equal(k, b' ', 'sla_kbj, jb1')
725727
T.assert_equal(j, 1, 'sla_kbj, j1')
726728
k, j = S.sla_kbj(0, e)
727-
T.assert_equal(k, 'B', 'sla_kbj, jb2')
729+
T.assert_equal(k, b'B', 'sla_kbj, jb2')
728730
T.assert_equal(j, 0, 'sla_kbj, j2')
729731
k, j = S.sla_kbj(1, e)
730-
T.assert_equal(k, 'B', 'sla_kbj, jb3')
732+
T.assert_equal(k, b'B', 'sla_kbj, jb3')
731733
T.assert_equal(j, 0, 'sla_kbj, j3')
732734
k, j = S.sla_kbj(2, e)
733-
T.assert_equal(k, 'J', 'sla_kbj, jb4')
735+
T.assert_equal(k, b'J', 'sla_kbj, jb4')
734736
T.assert_equal(j, 0, 'sla_kbj, j4')
735737
k, j = S.sla_kbj(3, e)
736-
T.assert_equal(k, ' ', 'sla_kbj, jb5')
738+
T.assert_equal(k, b' ', 'sla_kbj, jb5')
737739
T.assert_equal(j, 1, 'sla_kbj, j5')
738740
e = 2000
739741
k, j = S.sla_kbj(0, e)
740-
T.assert_equal(k, 'J', 'sla_kbj, jb6')
742+
T.assert_equal(k, b'J', 'sla_kbj, jb6')
741743
T.assert_equal(j, 0, 'sla_kbj, j6')
742744
k, j = S.sla_kbj(1, e)
743-
T.assert_equal(k, 'B', 'sla_kbj, jb7')
745+
T.assert_equal(k, b'B', 'sla_kbj, jb7')
744746
T.assert_equal(j, 0, 'sla_kbj, j7')
745747
k, j = S.sla_kbj(2, e)
746-
T.assert_equal(k, 'J', 'sla_kbj, jb8')
748+
T.assert_equal(k, b'J', 'sla_kbj, jb8')
747749
T.assert_equal(j, 0, 'sla_kbj, j8')
748750

749751
def testmap(self):
@@ -808,31 +810,31 @@ def testobs(self):
808810
n = 0
809811
c = 'MMT'
810812
c, name, w, p, h = S.sla_obs(n, c)
811-
T.assert_equal(c.strip(), 'MMT', 'sla_obs, 1/c')
812-
T.assert_equal(name.strip(), 'MMT 6.5m, Mt Hopkins', 'sla_obs, 1/name')
813+
T.assert_equal(c.strip(), b'MMT', 'sla_obs, 1/c')
814+
T.assert_equal(name.strip(), b'MMT 6.5m, Mt Hopkins', 'sla_obs, 1/name')
813815
T.assert_almost_equal(w, 1.935300584055477, 8, 'sla_obs, 1/w')
814816
T.assert_almost_equal(p, 0.5530735081550342238, 10, 'sla_obs, 1/p')
815817
T.assert_almost_equal(h, 2608, 10, 'sla_obs, 1/h')
816818
n = 61
817819
c = 20*' '
818820
c, name, w, p, h = S.sla_obs(n, c)
819-
T.assert_equal(c.strip(), 'KECK1', 'sla_obs, 2/c')
820-
T.assert_equal(name.strip(), 'Keck 10m Telescope #1', 'sla_obs, 2/name')
821+
T.assert_equal(c.strip(), b'KECK1', 'sla_obs, 2/c')
822+
T.assert_equal(name.strip(), b'Keck 10m Telescope #1', 'sla_obs, 2/name')
821823
T.assert_almost_equal(w, 2.713545757918895, 8, 'sla_obs, 2/w')
822824
T.assert_almost_equal(p, 0.3460280563536619, 8, 'sla_obs, 2/p')
823825
T.assert_almost_equal(h, 4160, 10, 'sla_obs, 2/h')
824826
n = 83
825827
c = 20*' '
826828
c, name, w, p, h = S.sla_obs(n, c)
827-
T.assert_equal(c.strip(), 'MAGELLAN2', 'sla_obs, 3/c')
828-
T.assert_equal(name.strip(), 'Magellan 2, 6.5m, Las Campanas', 'sla_obs, 3/name')
829+
T.assert_equal(c.strip(), b'MAGELLAN2', 'sla_obs, 3/c')
830+
T.assert_equal(name.strip(), b'Magellan 2, 6.5m, Las Campanas', 'sla_obs, 3/name')
829831
T.assert_almost_equal(w, 1.233819305534497, 8, 'sla_obs, 3/w')
830832
T.assert_almost_equal(p, -0.506389344359954, 8, 'sla_obs, 3/p')
831833
T.assert_almost_equal(h, 2408, 10, 'sla_obs, 3/h')
832834
n = 84
833835
c = 20*' '
834836
c, name, w, p, h = S.sla_obs(n, c)
835-
T.assert_equal(name.strip(), '?', 'sla_obs, 4/name')
837+
T.assert_equal(name.strip(), b'?', 'sla_obs, 4/name')
836838

837839
def testpa(self):
838840
T.assert_almost_equal(S.sla_pa(-1.567, 1.5123, 0.987),
@@ -1192,13 +1194,13 @@ def testsmat(self):
11921194
a = N.asarray([[2.22, 1.6578, 1.380522],
11931195
[1.6578, 1.380522, 1.22548578],
11941196
[1.380522, 1.22548578, 1.1356276122]],
1195-
order='Fortran')
1197+
order='F')
11961198
v = N.asarray([2.28625, 1.7128825, 1.429432225])
11971199
a, v, d, j, iw = S.sla_smat(a, v, N.empty(3, dtype=N.float32))
11981200
ans = N.asarray([[18.02550629769198, -52.16386644917280607, 34.37875949717850495],
11991201
[-52.16386644917280607, 168.1778099099805627, -118.0722869694232670],
12001202
[34.37875949717850495, -118.0722869694232670, 86.50307003740151262]],
1201-
order='Fortran')
1203+
order='F')
12021204
T.assert_array_almost_equal(a, ans, 2, 'sla_smat, a(3,3)')
12031205
ans = N.asarray([1.002346480763383, 0.03285594016974583489, 0.004760688414885247309])
12041206
T.assert_array_almost_equal(v, ans, 4, 'sla_smat, v(3)')
@@ -1216,7 +1218,7 @@ def testsvd(self):
12161218
nc = 7
12171219
m = 5
12181220
n = 4
1219-
a = N.empty((m, n), order='Fortran')
1221+
a = N.empty((m, n), order='F')
12201222
b = N.empty(m)
12211223
for ii in range(m):
12221224
val = 0.5 * (ii + 1.0)

0 commit comments

Comments
 (0)