def setUp(self): events.unsub_all() parser = AstromParser() directory_manager = LocalDirectoryWorkingContext( self._get_working_dir()) self.context = directory_manager progress_manager = LocalProgressManager(directory_manager) workunit_provider = WorkUnitProvider( tasks.get_suffix(self._get_task()), directory_manager, progress_manager, self._get_workunit_builder(parser, progress_manager)) self.workunit_provider = workunit_provider self.progress_manager = progress_manager self.singlet_download_manager = Mock(spec=AsynchronousDownloadManager) self.triplet_download_manager = Mock(spec=AsynchronousDownloadManager) self.image_manager = ImageManager(self.singlet_download_manager, self.triplet_download_manager) self.model = ValidationModel(self.workunit_provider, self.image_manager, None) self.custom_setup() self.model.start_work()
class GeneralModelTest(FileReadingTestCase, DirectoryCleaningTestCase): def setUp(self): events.unsub_all() parser = AstromParser() directory_manager = LocalDirectoryWorkingContext( self._get_working_dir()) self.context = directory_manager progress_manager = LocalProgressManager(directory_manager) workunit_provider = WorkUnitProvider( tasks.get_suffix(self._get_task()), directory_manager, progress_manager, self._get_workunit_builder(parser, progress_manager)) self.workunit_provider = workunit_provider self.progress_manager = progress_manager self.singlet_download_manager = Mock(spec=AsynchronousDownloadManager) self.triplet_download_manager = Mock(spec=AsynchronousDownloadManager) self.image_manager = ImageManager(self.singlet_download_manager, self.triplet_download_manager) self.model = ValidationModel(self.workunit_provider, self.image_manager, None) self.custom_setup() self.model.start_work() def custom_setup(self): pass def _get_task(self): raise NotImplementedError() def _get_working_dir(self): raise NotImplementedError() def get_directory_to_clean(self): return self._get_working_dir() def _get_workunit_builder(self, parser, progress_manager): raise NotImplementedError() def create_real_first_image(self, path="data/testimg.fits"): # Put a real fits image on the first source, first observation apcor = ApcorData.from_string("4 15 0.19 0.01") hdulist = fits.open(self.get_abs_path(path)) first_reading = self.model.get_current_workunit().get_sources( )[0].get_readings()[0] self.first_snapshot = SourceCutout(first_reading, hdulist, apcor) self.image_manager.on_singlet_image_loaded(self.first_snapshot)
class GeneralModelTest(FileReadingTestCase, DirectoryCleaningTestCase): def setUp(self): events.unsub_all() parser = AstromParser() directory_manager = LocalDirectoryWorkingContext(self._get_working_dir()) self.context = directory_manager progress_manager = LocalProgressManager(directory_manager) workunit_provider = WorkUnitProvider(tasks.get_suffix(self._get_task()), directory_manager, progress_manager, self._get_workunit_builder(parser, progress_manager)) self.workunit_provider = workunit_provider self.progress_manager = progress_manager self.singlet_download_manager = Mock(spec=AsynchronousDownloadManager) self.triplet_download_manager = Mock(spec=AsynchronousDownloadManager) self.image_manager = ImageManager(self.singlet_download_manager, self.triplet_download_manager) self.model = ValidationModel(self.workunit_provider, self.image_manager, None) self.custom_setup() self.model.start_work() def custom_setup(self): pass def _get_task(self): raise NotImplementedError() def _get_working_dir(self): raise NotImplementedError() def get_directory_to_clean(self): return self._get_working_dir() def _get_workunit_builder(self, parser, progress_manager): raise NotImplementedError() def create_real_first_image(self, path="data/testimg.fits"): # Put a real fits image on the first source, first observation apcor = ApcorData.from_string("4 15 0.19 0.01") hdulist = fits.open(self.get_abs_path(path)) first_reading = self.model.get_current_workunit().get_sources()[0].get_readings()[0] self.first_snapshot = SourceCutout( first_reading, hdulist, CoordinateConverter(0, 0), apcor) self.image_manager.on_singlet_image_loaded(self.first_snapshot)
def _create_image_manager(self): error_handler = DownloadErrorHandler(self) def read(slice_config): return config.read("CUTOUTS.%s" % slice_config) singlet_downloader = ImageCutoutDownloader( slice_rows=read("SINGLETS.SLICE_ROWS"), slice_cols=read("SINGLETS.SLICE_COLS")) singlet_download_manager = AsynchronousDownloadManager( singlet_downloader, error_handler) triplet_downloader = ImageCutoutDownloader( slice_rows=read("TRIPLETS.SLICE_ROWS"), slice_cols=read("TRIPLETS.SLICE_COLS")) triplet_download_manager = AsynchronousDownloadManager( triplet_downloader, error_handler) return ImageManager(singlet_download_manager, triplet_download_manager)
def setUp(self): events.unsub_all() parser = AstromParser() directory_manager = LocalDirectoryWorkingContext(self._get_working_dir()) self.context = directory_manager progress_manager = LocalProgressManager(directory_manager) workunit_provider = WorkUnitProvider(tasks.get_suffix(self._get_task()), directory_manager, progress_manager, self._get_workunit_builder(parser, progress_manager)) self.workunit_provider = workunit_provider self.progress_manager = progress_manager self.singlet_download_manager = Mock(spec=AsynchronousDownloadManager) self.triplet_download_manager = Mock(spec=AsynchronousDownloadManager) self.image_manager = ImageManager(self.singlet_download_manager, self.triplet_download_manager) self.model = ValidationModel(self.workunit_provider, self.image_manager, None) self.custom_setup() self.model.start_work()