def test_continue_writing_to_existing_file(self):
with open(self.get_abs_path(TEST_FILE_1), "rb") as fh:
expected_lines = fh.readlines()
def get_expected(num_lines):
return "".join(expected_lines[:num_lines])
astrom_data = self.parse(TEST_FILE_1)
source1 = astrom_data.sources[0]
source2 = astrom_data.sources[1]
assert_that(self.read_output(), equal_to(""))
self.writer.write_source(source1)
assert_that(self.read_output(), equal_to(get_expected(28)))
writer2 = StreamingAstromWriter(self.outputfile, self.sys_header)
writer2.write_source(source2)
assert_that(self.read_output(), equal_to(get_expected(32)))
def _create_results_writer(self, full_path, parsed_data):
output_filename = full_path.replace(tasks.get_suffix(tasks.CANDS_TASK),
tasks.get_suffix(tasks.REALS_TASK))
output_filehandle = open(output_filename, "a+b")
return StreamingAstromWriter(output_filehandle, parsed_data.sys_header)
def _create_writer(self):
filename = self.get_output_filename()
return StreamingAstromWriter(self.output_context.open(filename),
self.data.sys_header)
def setUp(self):
super(StreamingAstromWriterTest, self).setUp()
self.sys_header = {astrom.RMIN: 0.5, astrom.RMAX: 10.3,
astrom.ANGLE: -19.9, astrom.AWIDTH: 22.3}
self.writer = StreamingAstromWriter(self.outputfile, self.sys_header)
