def test_vis_wstack_iterator(self):
self.actualSetUp()
nchunks = len(list(vis_wstack_iter(self.vis, wstack=10.0)))
log.debug('Found %d chunks' % (nchunks))
assert nchunks > 1
for chunk, rows in enumerate(vis_wstack_iter(self.vis, wstack=10.0)):
assert len(rows)
visslice = create_visibility_from_rows(self.vis, rows)
assert numpy.sum(visslice.nvis) < self.vis.nvis
def test_vis_wstack_iterator(self):
self.actualSetUp()
nchunks = len(list(vis_wstack_iter(self.vis, wstack=10.0)))
log.debug('Found %d chunks' % (nchunks))
assert nchunks > 1
total_rows = 0
for chunk, rows in enumerate(vis_wstack_iter(self.vis, wstack=10.0)):
assert len(rows)
visslice = create_visibility_from_rows(self.vis, rows)
total_rows += visslice.nvis
assert numpy.sum(visslice.nvis) < self.vis.nvis
assert total_rows == self.vis.nvis, "Total rows iterated %d, Original rows %d" % (
total_rows, self.vis.nvis)