Source code for cis_interface.drivers.tests.test_AsciiTableInputDriver

import nose.tools as nt
import cis_interface.drivers.tests.test_AsciiFileInputDriver as parent
import cis_interface.drivers.tests.test_FileInputDriver as super_parent


[docs]class TestAsciiTableInputParam(parent.TestAsciiFileInputParam): r"""Test parameters for AsciiTableInputDriver. Attributes (in addition to parent class's): - """ def __init__(self, *args, **kwargs): super(TestAsciiTableInputParam, self).__init__(*args, **kwargs) self.driver = 'AsciiTableInputDriver' self.attr_list += ['file', 'as_array']
[docs]class TestAsciiTableInputDriverNoStart(TestAsciiTableInputParam, parent.TestAsciiFileInputDriverNoStart): r"""Test runner for AsciiTableInputDriver. Attributes (in addition to parent class's): - """ pass
[docs]class TestAsciiTableInputDriver(TestAsciiTableInputParam, parent.TestAsciiFileInputDriver): r"""Test runner for AsciiTableInputDriver. Attributes (in addition to parent class's): - """
[docs] def assert_before_stop(self): r"""Assertions to make before stopping the driver instance.""" super(super_parent.TestFileInputDriver, self).assert_before_stop() elapsed = 0.0 while self.instance.n_ipc_msg == 0 and (elapsed <= self.instance.timeout): self.instance.sleep() elapsed += self.instance.sleeptime data = self.instance.ipc_recv() nt.assert_equal(data, self.fmt_str) iline = 0 while True: data = self.instance.ipc_recv_nolimit() if (data is None) or (data == self.instance.eof_msg): break if len(data) > 0: while self.file_lines[iline].startswith(self.comment): iline += 1 # pragma: no cover nt.assert_equal(data, self.file_lines[iline]) iline += 1 nt.assert_equal(len(self.file_lines), iline)
[docs]class TestAsciiTableInputDriver_Array(TestAsciiTableInputParam, parent.TestAsciiFileInputDriver): r"""Test runner for AsciiTableInputDriver with array input. Attributes (in addition to parent class's): - """ def __init__(self, *args, **kwargs): super(TestAsciiTableInputDriver_Array, self).__init__(*args, **kwargs) self.args = {'filepath': self.filepath, 'as_array': True}
[docs] def assert_before_stop(self): r"""Assertions to make before stopping the driver instance.""" super(super_parent.TestFileInputDriver, self).assert_before_stop() elapsed = 0.0 while self.instance.n_ipc_msg == 0 and (elapsed <= self.instance.timeout): self.instance.sleep() elapsed += self.instance.sleeptime data = self.instance.ipc_recv() nt.assert_equal(data, self.fmt_str) data = self.instance.recv_wait_nolimit() nt.assert_equal(data, self.file_bytes)