| 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
|
|---|
