| 1 |
schorsch |
1.1 |
# -*- coding: utf-8 -*- |
| 2 |
|
|
from __future__ import division, print_function, unicode_literals |
| 3 |
|
|
|
| 4 |
|
|
import struct |
| 5 |
|
|
import operator |
| 6 |
|
|
import unittest |
| 7 |
|
|
from functools import reduce |
| 8 |
|
|
|
| 9 |
|
|
import testsupport as ts |
| 10 |
|
|
from pyradlib import lcompare |
| 11 |
|
|
from pyradlib.pyrad_proc import PIPE, Error, ProcMixin |
| 12 |
|
|
|
| 13 |
|
|
|
| 14 |
|
|
class PyradprocTestCase(unittest.TestCase, ProcMixin): |
| 15 |
|
|
|
| 16 |
|
|
def _runit(self, cmdl, actstrl, _in=None, indata=None, nl=False): |
| 17 |
|
|
try: |
| 18 |
|
|
procs = [] |
| 19 |
|
|
if isinstance(cmdl[0], (type(b''), type(u''))): |
| 20 |
|
|
proc = self.call_one(cmdl, actstrl, _in=_in, out=PIPE, |
| 21 |
|
|
universal_newlines=nl) |
| 22 |
|
|
procs = [proc] |
| 23 |
|
|
elif len(cmdl) == 2: |
| 24 |
|
|
procs = self.call_two(cmdl[0], cmdl[1], actstrl[0], actstrl[1], |
| 25 |
|
|
_in=_in, out=PIPE, universal_newlines=nl) |
| 26 |
|
|
else: |
| 27 |
|
|
procs = self.call_many(cmdl, actstrl, _in=_in, out=PIPE, |
| 28 |
|
|
universal_newlines=nl) |
| 29 |
|
|
if indata: # only for small amounts of data |
| 30 |
|
|
procs[0].stdin.write(indata) |
| 31 |
|
|
procs[0].stdin.close() |
| 32 |
|
|
res = procs[-1].stdout.read() |
| 33 |
|
|
else: |
| 34 |
|
|
res, eres = procs[-1].communicate() |
| 35 |
|
|
if eres: print('Errors', eres) # XXX |
| 36 |
|
|
except Error as e: |
| 37 |
|
|
raise |
| 38 |
|
|
self.fail('%s [%s]' % (str(e), self.qjoin(cmdl))) |
| 39 |
|
|
finally: |
| 40 |
|
|
for proc in procs: |
| 41 |
|
|
proc.wait() |
| 42 |
|
|
if nl: return filter(None, lcompare.split_rad(res)) |
| 43 |
|
|
return res |
| 44 |
|
|
|
| 45 |
|
|
|
| 46 |
|
|
def test_call_1_text(self): |
| 47 |
|
|
for n in (5, 10, 55, 200,1328,1329,1330,1331,1332, 2000, 5000,99999): |
| 48 |
|
|
exp = range(n) |
| 49 |
|
|
cntcmd = ['cnt', str(n)] |
| 50 |
|
|
res = self._runit(cntcmd, 'run one process', nl=True) |
| 51 |
|
|
resl = [r[0] for r in res] |
| 52 |
|
|
try: lcompare.lcompare(resl, exp) |
| 53 |
|
|
except lcompare.error as e: |
| 54 |
|
|
print(resl, exp) |
| 55 |
|
|
self.fail(('call_one_text n=%d -- ' % n) +str(e)) |
| 56 |
|
|
|
| 57 |
|
|
def test_call_1_bin(self): |
| 58 |
|
|
exp = (2160, 8640, 4320, 1080, 7560) |
| 59 |
|
|
cntcmd = ['total', '-of'] |
| 60 |
|
|
tfn = ts.datafile('histo.dat') |
| 61 |
|
|
res = self._runit(cntcmd, 'run one process', _in=tfn, nl=False) |
| 62 |
|
|
resl = struct.unpack('f'*len(exp), res) |
| 63 |
|
|
try: lcompare.lcompare(resl, exp) |
| 64 |
|
|
except lcompare.error as e: |
| 65 |
|
|
self.fail(('call_one_text n=%d -- ' % n) +str(e)) |
| 66 |
|
|
|
| 67 |
|
|
def test_call_1_noexe(self): |
| 68 |
|
|
# let's hope a program with that name doesn't exist |
| 69 |
|
|
cmd = ['iswaxubdzkcgrhmltqvyfepojn', '-x', '-y', '-z'] |
| 70 |
|
|
self.assertRaises(Error, self._runit, cmd, 'run one process') |
| 71 |
|
|
|
| 72 |
|
|
def test_call_2_text(self): |
| 73 |
|
|
# with values higher than 44721, total will return a float in e-format. |
| 74 |
|
|
for n in (5, 10, 55, 200,1328,1329,1330,1331,1332, 2000, 5000, |
| 75 |
|
|
44721, 44722, 99999): |
| 76 |
|
|
exp = [reduce(operator.add, range(n))] |
| 77 |
|
|
cntcmd = ['cnt', str(n)] |
| 78 |
|
|
totcmd = ['total'] |
| 79 |
|
|
resl = self._runit([cntcmd, totcmd], ['run p1','run p2'], nl=True) |
| 80 |
|
|
try: lcompare.lcompare(resl, [exp]) |
| 81 |
|
|
except lcompare.error as e: |
| 82 |
|
|
self.fail(('call_one_text n=%d -- ' % n) +str(e)) |
| 83 |
|
|
|
| 84 |
|
|
def test_call_2_bin(self): |
| 85 |
|
|
for nn in ((1,1,1), (5,5,5), (3,9,5,2,8)): |
| 86 |
|
|
totalrows = reduce(operator.mul, nn) |
| 87 |
|
|
exp = [reduce(operator.add, range(n))*(totalrows/n) for n in nn] |
| 88 |
|
|
cntcmd = ['cnt'] + [str(n) for n in nn] |
| 89 |
|
|
totcmd = ['total', '-of'] |
| 90 |
|
|
res = self._runit([cntcmd, totcmd], ['run p1','run p2']) |
| 91 |
|
|
resl = struct.unpack('f'*len(exp), res) |
| 92 |
|
|
try: lcompare.lcompare(resl, exp) |
| 93 |
|
|
except lcompare.error as e: |
| 94 |
|
|
self.fail(('call_one_text n=%d -- ' % n) +str(e)) |
| 95 |
|
|
|
| 96 |
|
|
def test_call_many_text(self): |
| 97 |
|
|
data = b'''void polygon test 0 0 9 0 0 0 1 0 0 0 1 0''' |
| 98 |
|
|
exp = (['#', 'xform', '-s', 0.5], |
| 99 |
|
|
['#', 'xform', '-t', 0, 0, 2], |
| 100 |
|
|
['#', 'xform', '-ry', -20], |
| 101 |
|
|
['#', 'xform', '-rx', 30], |
| 102 |
|
|
['void', 'polygon', 'test'], |
| 103 |
|
|
[0], [0], [9], |
| 104 |
|
|
[0, 0, 1], |
| 105 |
|
|
[0.469846310393, 0, 1.17101007167], |
| 106 |
|
|
[-0.0855050358315, 0.433012701892, 1.23492315519]) |
| 107 |
|
|
cmd0 = ['xform', '-rx', '30'] |
| 108 |
|
|
cmd1 = ['xform', '-ry', '-20'] |
| 109 |
|
|
cmd2 = ['xform', '-t', '0', '0', '2'] |
| 110 |
|
|
cmd3 = ['xform', '-s', '0.5'] |
| 111 |
|
|
cmdl = [cmd0, cmd1, cmd2, cmd3] |
| 112 |
|
|
resl = self._runit(cmdl, 'run many processes', indata=data, |
| 113 |
|
|
_in=PIPE, nl=True) |
| 114 |
|
|
try: lcompare.lcompare(resl, exp) |
| 115 |
|
|
except lcompare.error as e: |
| 116 |
|
|
print(resl, exp) |
| 117 |
|
|
self.fail(('call_one_text n=%d -- ' % n) +str(e)) |
| 118 |
|
|
|
| 119 |
|
|
def test_call_many_bin(self): |
| 120 |
|
|
exp = (15, 40, 45, 48, 0) |
| 121 |
|
|
cmd0 = ['cnt', '3', '4', '5', '1'] |
| 122 |
|
|
cmd1 = ['histo', '1', '5'] |
| 123 |
|
|
cmd2 = ['total', '-of'] |
| 124 |
|
|
cmdl = [cmd0, cmd1, cmd2] |
| 125 |
|
|
res = self._runit(cmdl, 'run many processes') |
| 126 |
|
|
resl = struct.unpack('f'*len(exp), res) |
| 127 |
|
|
try: lcompare.lcompare(resl, exp) |
| 128 |
|
|
except lcompare.error as e: |
| 129 |
|
|
print(resl, exp) |
| 130 |
|
|
self.fail(('call_one_text -- ') +str(e)) |
| 131 |
|
|
|
| 132 |
|
|
|
| 133 |
|
|
# vi: set ts=4 sw=4 : |
