| 1 |
|
| 2 |
|
| 3 |
import os |
| 4 |
import math |
| 5 |
import string |
| 6 |
import unittest |
| 7 |
|
| 8 |
from unit_tools import support |
| 9 |
from unit_tools import lcompare |
| 10 |
|
| 11 |
_bindir = None |
| 12 |
|
| 13 |
class CntTestCase(unittest.TestCase): |
| 14 |
def setUp(self): |
| 15 |
if _bindir: |
| 16 |
self.oldpath = os.environ['PATH'] |
| 17 |
os.environ['PATH'] = _bindir |
| 18 |
|
| 19 |
def tearDown(self): |
| 20 |
if _bindir: |
| 21 |
os.environ['PATH'] = self.oldpath |
| 22 |
|
| 23 |
def test_1(self): |
| 24 |
cmd = 'cnt 5' |
| 25 |
res0 = os.popen(cmd).read() |
| 26 |
res = map(string.strip,string.split(res0, '\n')) |
| 27 |
exp = [0, 1, 2, 3, 4, ''] |
| 28 |
try: lcompare.lcompare(res, exp) |
| 29 |
except lcompare.error, e: self.fail(str(e)) |
| 30 |
|
| 31 |
def test_2(self): |
| 32 |
cmd = 'cnt 3 2' |
| 33 |
res0 = os.popen(cmd).read() |
| 34 |
res = map(string.split,string.split(res0, '\n')) |
| 35 |
exp = [[0,0], [0,1], [1,0], [1,1], [2,0], [2,1], []] |
| 36 |
try: lcompare.llcompare(res, exp) |
| 37 |
except lcompare.error, e: self.fail(str(e)) |
| 38 |
|
| 39 |
def test_3(self): |
| 40 |
cmd = 'cnt 3 2 3' |
| 41 |
res0 = os.popen(cmd).read() |
| 42 |
res = map(string.split,string.split(res0, '\n')) |
| 43 |
exp = [[0,0,0],[0,0,1],[0,0,2], |
| 44 |
[0,1,0],[0,1,1],[0,1,2], |
| 45 |
[1,0,0],[1,0,1],[1,0,2], |
| 46 |
[1,1,0],[1,1,1],[1,1,2], |
| 47 |
[2,0,0],[2,0,1],[2,0,2], |
| 48 |
[2,1,0],[2,1,1],[2,1,2], |
| 49 |
[]] |
| 50 |
try: lcompare.llcompare(res, exp) |
| 51 |
except lcompare.error, e: self.fail(str(e)) |
| 52 |
|
| 53 |
|
| 54 |
def main(bindir=None): |
| 55 |
global _bindir |
| 56 |
_bindir=bindir |
| 57 |
support.run_case(CntTestCase) |
| 58 |
|
| 59 |
if __name__ == '__main__': |
| 60 |
main() |
| 61 |
|
| 62 |
# vi: set ts=4 sw=4 : |
