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)