def setUp(self):
source = Mock(spec=astrom.Source)
source.num_readings.return_value = 3
def mock_hdulist():
return MagicMock(spec=HDUList)
grid = CutoutGrid(source)
self.hdulist00 = mock_hdulist()
self.hdulist01 = mock_hdulist()
self.hdulist02 = mock_hdulist()
self.hdulist10 = mock_hdulist()
self.hdulist11 = mock_hdulist()
self.hdulist12 = mock_hdulist()
self.hdulist20 = mock_hdulist()
self.hdulist21 = mock_hdulist()
self.hdulist22 = mock_hdulist()
def mock_cutout(hdulist):
cutout = Mock(spec=SourceCutout)
cutout.hdulist = hdulist
return cutout
grid.add_cutout(mock_cutout(self.hdulist00), 0, 0)
grid.add_cutout(mock_cutout(self.hdulist01), 0, 1)
grid.add_cutout(mock_cutout(self.hdulist02), 0, 2)
grid.add_cutout(mock_cutout(self.hdulist10), 1, 0)
grid.add_cutout(mock_cutout(self.hdulist11), 1, 1)
grid.add_cutout(mock_cutout(self.hdulist12), 1, 2)
grid.add_cutout(mock_cutout(self.hdulist20), 2, 0)
grid.add_cutout(mock_cutout(self.hdulist21), 2, 1)
grid.add_cutout(mock_cutout(self.hdulist22), 2, 2)
self.grid = grid
def setUp(self):
self.source = Mock(spec=astrom.Source)
self.source.num_readings.return_value = 2
self.grid = CutoutGrid(self.source)
self.cutout00 = Mock(spec=SourceCutout)
self.cutout01 = Mock(spec=SourceCutout)
self.cutout10 = Mock(spec=SourceCutout)
self.cutout11 = Mock(spec=SourceCutout)
self.grid.add_cutout(self.cutout00, 0, 0)
self.grid.add_cutout(self.cutout01, 0, 1)
self.grid.add_cutout(self.cutout10, 1, 0)
self.grid.add_cutout(self.cutout11, 1, 1)
def test_is_filled(self):
source = Mock(spec=astrom.Source)
source.num_readings.return_value = 2
grid = CutoutGrid(source)
def add_cutout(frame_index, time_index):
grid.add_cutout(Mock(spec=SourceCutout), frame_index, time_index)
assert_that(grid.is_filled(), equal_to(False))
add_cutout(0, 0)
assert_that(grid.is_filled(), equal_to(False))
add_cutout(1, 0)
assert_that(grid.is_filled(), equal_to(False))
add_cutout(1, 1)
assert_that(grid.is_filled(), equal_to(False))
add_cutout(0, 1)
assert_that(grid.is_filled(), equal_to(True))
