def test_sampleUniformly(self): hns = sampleUniformly(4, [(0, 1), (0, 1)]) self.assertEqual(hns, [[0, 0], [1, 0], [0, 1], [1, 1]]) hns = sampleUniformly(4, [(0, 4), (0, 4)]) self.assertEqual(hns, [[1, 1], [3, 1], [1, 3], [3, 3]]) hns = sampleUniformly(4, [(0, 4), (0, 8)]) self.assertEqual(hns, [[1, 3], [3, 3], [1, 5], [3, 5]]) hns = sampleUniformly(3, [(0, 4), (0, 8)]) self.assertEqual(hns, [[2, 2], [2, 4], [2, 6]])
def test_sampleHolesUniformly(self): ncols = 1144 nrows = 1024 bounds = [(64, ncols), (64, nrows)] expected = [xy_to_hn(x, y) for x,y in sampleUniformly(100, bounds)] all_xy = quadratic_expand([range(64, ncols), range(64, nrows)]) all_holes = [xy_to_hn(x, y) for x, y in all_xy] samples = sampleHolesUniformly(100, all_holes) self.assertEqual(expected, samples)