[1864] | 1 | import os |
---|
| 2 | import shutil |
---|
| 3 | from nose.tools import * |
---|
| 4 | from asap import linecatalog |
---|
| 5 | |
---|
| 6 | FNAME = "12mmlines.dat" |
---|
| 7 | |
---|
| 8 | class TestLinecatalog: |
---|
| 9 | @classmethod |
---|
| 10 | def setupClass(self): |
---|
| 11 | f = file(FNAME, "w") |
---|
| 12 | txt = """\ |
---|
| 13 | He88B 18669.75 0.0 1.0 |
---|
| 14 | H87B 19304.69 0.0 2 |
---|
| 15 | He87B 19312.55 0.0 3 |
---|
| 16 | U 19416.0 0.0 0 |
---|
| 17 | """ |
---|
| 18 | f.write(txt) |
---|
| 19 | f.close() |
---|
| 20 | |
---|
| 21 | @classmethod |
---|
| 22 | def teardownClass(self): |
---|
| 23 | os.remove(FNAME) |
---|
| 24 | |
---|
| 25 | def setup(self): |
---|
| 26 | self.lc = linecatalog(FNAME) |
---|
| 27 | |
---|
| 28 | def test_init(self): |
---|
| 29 | pass |
---|
| 30 | |
---|
| 31 | def test_len(self): |
---|
| 32 | assert_equals(len(self.lc), 4) |
---|
| 33 | |
---|
| 34 | def test_set_name_pattern(self): |
---|
| 35 | self.lc.set_name("H*") |
---|
| 36 | assert_equals(len(self.lc), 3) |
---|
| 37 | |
---|
| 38 | @raises(RuntimeError) |
---|
| 39 | def test_fail_set_name_pattern(self): |
---|
| 40 | self.lc.set_name("Z*") |
---|
| 41 | |
---|
| 42 | def test_set_name_regex(self): |
---|
| 43 | self.lc.set_name("^H[0-9]+B$", 'regex') |
---|
| 44 | assert_equals(len(self.lc), 1) |
---|
| 45 | |
---|
| 46 | @raises(RuntimeError) |
---|
| 47 | def test_fail_set_name_regex(self): |
---|
| 48 | self.lc.set_name("^Z[0-9]+B$", 'regex') |
---|
| 49 | |
---|
| 50 | def test_reset(self): |
---|
| 51 | self.lc.set_name("^H[0-9]+B$", 'regex') |
---|
| 52 | self.lc.reset() |
---|
| 53 | assert_equals(len(self.lc), 4) |
---|
| 54 | |
---|
| 55 | def test_set_frequency_limits(self): |
---|
| 56 | for args in [(1.0, 19.0, 'GHz', 1), |
---|
| 57 | (1000.0, 19000.0, 'MHz', 1), |
---|
| 58 | (19.30, 19.4, 'GHz', 2), |
---|
| 59 | ]: |
---|
| 60 | yield self.freq_limits, args |
---|
| 61 | |
---|
| 62 | def freq_limits(self, args): |
---|
| 63 | self.lc.set_frequency_limits(*args[:-1]) |
---|
| 64 | assert_equals(len(self.lc), args[-1]) |
---|
| 65 | |
---|
| 66 | @raises(RuntimeError) |
---|
| 67 | def test_fails_set_frequency_limits(self): |
---|
| 68 | self.lc.set_frequency_limits(20, 30) |
---|
| 69 | |
---|
| 70 | def test_set_strength_limits(self): |
---|
| 71 | self.lc.set_strength_limits(0.5, 2.5) |
---|
| 72 | assert_equals(len(self.lc), 2) |
---|
| 73 | |
---|
| 74 | @raises(RuntimeError) |
---|
| 75 | def test_fails_set_strength_limits(self): |
---|
| 76 | self.lc.set_strength_limits(10, 20) |
---|
| 77 | |
---|
| 78 | |
---|
| 79 | def test_get_row(self): |
---|
| 80 | r = self.lc.get_row(0) |
---|
| 81 | refrow = {'name': 'He88B', 'value': 18669.75} |
---|
| 82 | assert_equals(r, refrow) |
---|
| 83 | |
---|
| 84 | @raises(IndexError) |
---|
| 85 | def test_fail_get_row(self): |
---|
| 86 | r = self.lc.get_row(-1) |
---|
| 87 | |
---|
| 88 | def test_get_item(self): |
---|
| 89 | r = self.lc[0] |
---|
| 90 | refrow = {'name': 'He88B', 'value': 18669.75} |
---|
| 91 | assert_equals(r, refrow) |
---|
| 92 | |
---|
| 93 | def test_save(self): |
---|
| 94 | outname = 'tmp_lincatalog.tbl' |
---|
| 95 | self.lc.save(outname) |
---|
| 96 | lc = linecatalog(outname) |
---|
| 97 | lctxt = str(lc) |
---|
| 98 | origlctxt = str(self.lc) |
---|
| 99 | lc = None |
---|
| 100 | shutil.rmtree(outname, True) |
---|
| 101 | assert lctxt == origlctxt |
---|