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)
