def test_processed_uniform_probabilities(self):
sampler = UniformSampler(5)
probabilities = sampler.get_probability_map(self.sample)
probabilities = sampler.process_probability_map(probabilities)
fixtures = np.zeros_like(probabilities)
# Other positions cannot be patch centers
fixtures[2:-2, 2:-2, 2:-2] = probabilities[2, 2, 2]
self.assertAlmostEqual(probabilities.sum(), 1)
assert np.equal(probabilities, fixtures).all()
def test_uniform_probabilities(self):
sampler = UniformSampler(5)
probabilities = sampler.get_probability_map(self.sample)
fixtures = torch.ones_like(probabilities)
assert torch.all(probabilities.eq(fixtures))