class SourceSnapshotTest(FileReadingTestCase): def setUp(self): test_file = "data/image_reading/realstest2.measure3.reals.astrom" astrom_data = AstromParser().parse(self.get_abs_path(test_file)) self.reading = astrom_data.get_sources()[0].get_reading(0) self.original_observed_x = self.reading.x self.original_observed_y = self.reading.y self.offset_x = 10 self.offset_y = 15 coordinate_converter = CoordinateConverter(self.offset_x, self.offset_y) self.undertest = SourceCutout(self.reading, Mock()) self.mock_update_ra_dec = Mock() self.undertest._update_ra_dec = self.mock_update_ra_dec def assert_update_ra_dec_call_count_equals(self, expected_calls): assert_that(self.mock_update_ra_dec.call_count, equal_to(expected_calls)) def test_xy_coordinate_systems(self): assert_that(self.undertest.observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y)) expected_pixel_x = self.original_observed_x - self.offset_x expected_pixel_y = self.original_observed_y - self.offset_y assert_that(self.undertest.pixel_x, equal_to(expected_pixel_x)) assert_that(self.undertest.pixel_y, equal_to(expected_pixel_y)) assert_that( self.undertest.observed_source_point, equal_to((self.original_observed_x, self.original_observed_y))) assert_that(self.undertest.pixel_source_point, equal_to((expected_pixel_x, expected_pixel_y))) def test_xy_adjusted(self): assert_that(self.undertest.observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y)) assert_that(self.undertest.is_adjusted(), equal_to(False)) diff_x = 2.5 diff_y = 3.5 new_pixel_x = self.undertest.pixel_x + diff_x new_pixel_y = self.undertest.pixel_y + diff_y self.undertest.update_pixel_location((new_pixel_x, new_pixel_y)) assert_that(self.undertest.pixel_x, equal_to(new_pixel_x)) assert_that(self.undertest.pixel_y, equal_to(new_pixel_y)) assert_that(self.undertest.observed_x, equal_to(self.original_observed_x + diff_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y + diff_y)) assert_that(self.undertest.is_adjusted(), equal_to(True)) def test_update_xy_updates_ra_dec(self): self.assert_update_ra_dec_call_count_equals(0) self.undertest.update_pixel_location((665.0, 3215.0)) _ = self.undertest.dec self.assert_update_ra_dec_call_count_equals(1) def test_ra_dec_updates_lazilly(self): self.assert_update_ra_dec_call_count_equals(0) self.undertest.update_pixel_location((665.0, 3215.0)) self.assert_update_ra_dec_call_count_equals(0) self.undertest.update_pixel_location((665.0, 3218.0)) self.assert_update_ra_dec_call_count_equals(0) _ = self.undertest.ra self.assert_update_ra_dec_call_count_equals(1) _ = self.undertest.dec self.assert_update_ra_dec_call_count_equals(1) self.undertest.update_pixel_location((667.2, 3216.6)) self.assert_update_ra_dec_call_count_equals(1) _ = self.undertest.dec _ = self.undertest.ra self.assert_update_ra_dec_call_count_equals(2) def test_original_xy(self): assert_that(self.undertest.observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y)) assert_that(self.undertest.original_observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.original_observed_y, equal_to(self.original_observed_y)) diff_x = 2.5 diff_y = 3.5 new_pixel_x = self.undertest.pixel_x + diff_x new_pixel_y = self.undertest.pixel_y + diff_y self.undertest.update_pixel_location((new_pixel_x, new_pixel_y)) assert_that(self.undertest.observed_x, equal_to(self.original_observed_x + diff_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y + diff_y)) assert_that(self.undertest.original_observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.original_observed_y, equal_to(self.original_observed_y)) def test_reset_location(self): original_pixel_x, original_pixel_y = self.undertest.pixel_source_point diff_x = 2.5 diff_y = 3.5 new_pixel_x = self.undertest.pixel_x + diff_x new_pixel_y = self.undertest.pixel_y + diff_y self.undertest.update_pixel_location((new_pixel_x, new_pixel_y)) assert_that(self.undertest.observed_x, equal_to(self.original_observed_x + diff_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y + diff_y)) assert_that(self.undertest.pixel_x, equal_to(new_pixel_x)) assert_that(self.undertest.pixel_y, equal_to(new_pixel_y)) self.undertest.reset_source_location() assert_that(self.undertest.observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y)) assert_that(self.undertest.pixel_x, equal_to(original_pixel_x)) assert_that(self.undertest.pixel_y, equal_to(original_pixel_y)) def test_reset_location_is_adjusted_false(self): assert_that(self.undertest.is_adjusted(), equal_to(False)) self.undertest.update_pixel_location((100, 100)) assert_that(self.undertest.is_adjusted(), equal_to(True)) self.undertest.reset_source_location() assert_that(self.undertest.is_adjusted(), equal_to(False)) def test_reset_location_makes_data_stale(self): self.assert_update_ra_dec_call_count_equals(0) self.undertest.update_pixel_location((665.0, 3215.0)) _ = self.undertest.ra self.assert_update_ra_dec_call_count_equals(1) self.undertest.reset_source_location() self.assert_update_ra_dec_call_count_equals(1) _ = self.undertest.ra self.assert_update_ra_dec_call_count_equals(2)
class SourceSnapshotTest(FileReadingTestCase): def setUp(self): test_file = "data/image_reading/realstest2.measure3.reals.astrom" astrom_data = AstromParser().parse(self.get_abs_path(test_file)) self.reading = astrom_data.get_sources()[0].get_reading(0) self.original_observed_x = self.reading.x self.original_observed_y = self.reading.y self.offset_x = 10 self.offset_y = 15 coordinate_converter = CoordinateConverter(self.offset_x, self.offset_y) self.undertest = SourceCutout(self.reading, Mock(), coordinate_converter) self.mock_update_ra_dec = Mock() self.undertest._update_ra_dec = self.mock_update_ra_dec def assert_update_ra_dec_call_count_equals(self, expected_calls): assert_that(self.mock_update_ra_dec.call_count, equal_to(expected_calls)) def test_xy_coordinate_systems(self): assert_that(self.undertest.observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y)) expected_pixel_x = self.original_observed_x - self.offset_x expected_pixel_y = self.original_observed_y - self.offset_y assert_that(self.undertest.pixel_x, equal_to(expected_pixel_x)) assert_that(self.undertest.pixel_y, equal_to(expected_pixel_y)) assert_that(self.undertest.observed_source_point, equal_to((self.original_observed_x, self.original_observed_y))) assert_that(self.undertest.pixel_source_point, equal_to((expected_pixel_x, expected_pixel_y))) def test_xy_adjusted(self): assert_that(self.undertest.observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y)) assert_that(self.undertest.is_adjusted(), equal_to(False)) diff_x = 2.5 diff_y = 3.5 new_pixel_x = self.undertest.pixel_x + diff_x new_pixel_y = self.undertest.pixel_y + diff_y self.undertest.update_pixel_location((new_pixel_x, new_pixel_y)) assert_that(self.undertest.pixel_x, equal_to(new_pixel_x)) assert_that(self.undertest.pixel_y, equal_to(new_pixel_y)) assert_that(self.undertest.observed_x, equal_to(self.original_observed_x + diff_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y + diff_y)) assert_that(self.undertest.is_adjusted(), equal_to(True)) def test_update_xy_updates_ra_dec(self): self.assert_update_ra_dec_call_count_equals(0) self.undertest.update_pixel_location((665.0, 3215.0)) _ = self.undertest.dec self.assert_update_ra_dec_call_count_equals(1) def test_ra_dec_updates_lazilly(self): self.assert_update_ra_dec_call_count_equals(0) self.undertest.update_pixel_location((665.0, 3215.0)) self.assert_update_ra_dec_call_count_equals(0) self.undertest.update_pixel_location((665.0, 3218.0)) self.assert_update_ra_dec_call_count_equals(0) _ = self.undertest.ra self.assert_update_ra_dec_call_count_equals(1) _ = self.undertest.dec self.assert_update_ra_dec_call_count_equals(1) self.undertest.update_pixel_location((667.2, 3216.6)) self.assert_update_ra_dec_call_count_equals(1) _ = self.undertest.dec _ = self.undertest.ra self.assert_update_ra_dec_call_count_equals(2) def test_original_xy(self): assert_that(self.undertest.observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y)) assert_that(self.undertest.original_observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.original_observed_y, equal_to(self.original_observed_y)) diff_x = 2.5 diff_y = 3.5 new_pixel_x = self.undertest.pixel_x + diff_x new_pixel_y = self.undertest.pixel_y + diff_y self.undertest.update_pixel_location((new_pixel_x, new_pixel_y)) assert_that(self.undertest.observed_x, equal_to(self.original_observed_x + diff_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y + diff_y)) assert_that(self.undertest.original_observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.original_observed_y, equal_to(self.original_observed_y)) def test_reset_location(self): original_pixel_x, original_pixel_y = self.undertest.pixel_source_point diff_x = 2.5 diff_y = 3.5 new_pixel_x = self.undertest.pixel_x + diff_x new_pixel_y = self.undertest.pixel_y + diff_y self.undertest.update_pixel_location((new_pixel_x, new_pixel_y)) assert_that(self.undertest.observed_x, equal_to(self.original_observed_x + diff_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y + diff_y)) assert_that(self.undertest.pixel_x, equal_to(new_pixel_x)) assert_that(self.undertest.pixel_y, equal_to(new_pixel_y)) self.undertest.reset_source_location() assert_that(self.undertest.observed_x, equal_to(self.original_observed_x)) assert_that(self.undertest.observed_y, equal_to(self.original_observed_y)) assert_that(self.undertest.pixel_x, equal_to(original_pixel_x)) assert_that(self.undertest.pixel_y, equal_to(original_pixel_y)) def test_reset_location_is_adjusted_false(self): assert_that(self.undertest.is_adjusted(), equal_to(False)) self.undertest.update_pixel_location((100, 100)) assert_that(self.undertest.is_adjusted(), equal_to(True)) self.undertest.reset_source_location() assert_that(self.undertest.is_adjusted(), equal_to(False)) def test_reset_location_makes_data_stale(self): self.assert_update_ra_dec_call_count_equals(0) self.undertest.update_pixel_location((665.0, 3215.0)) _ = self.undertest.ra self.assert_update_ra_dec_call_count_equals(1) self.undertest.reset_source_location() self.assert_update_ra_dec_call_count_equals(1) _ = self.undertest.ra self.assert_update_ra_dec_call_count_equals(2)