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