def test_generate_data_with_gaussian_background(self):
BackgroundGenerator.gaussian_noise(64, 128).save('tests/out/gaussian_background.jpg')
self.assertTrue(
md5('tests/out/gaussian_background.jpg') == md5('tests/expected_results/gaussian_background.jpg')
)
os.remove('tests/out/gaussian_background.jpg')
def test_generate_data_with_white_background(self):
BackgroundGenerator.plain_white(64, 128).save('tests/out/white_background.jpg')
self.assertTrue(
md5('tests/out/white_background.jpg') == md5('tests/expected_results/white_background.jpg')
)
os.remove('tests/out/white_background.jpg')
def test_generate_data_with_quasicrystal_background(self):
bkgd = BackgroundGenerator.quasicrystal(64, 128)
self.assertTrue(len(bkgd.histogram()) > 20 and bkgd.size == (128, 64))
