19 lines
789 B
Python
19 lines
789 B
Python
from unittest import TestCase
|
|
from esbo_etc.lib.helpers import rasterizeCircle
|
|
import numpy as np
|
|
|
|
|
|
class Test(TestCase):
|
|
def test_rasterize_circle(self):
|
|
grid = np.zeros((8, 8))
|
|
circ = rasterizeCircle(grid, 2.6, 4.5, 3.8)
|
|
circ_ex = np.array([[0., 0., 0., 0., 0., 0., 0., 0.],
|
|
[0., 0., 0., 0., 1., 1., 0., 0.],
|
|
[0., 0., 0., 1., 1., 1., 1., 0.],
|
|
[0., 0., 1., 1., 1., 1., 1., 1.],
|
|
[0., 0., 1., 1., 1., 1., 1., 1.],
|
|
[0., 0., 1., 1., 1., 1., 1., 1.],
|
|
[0., 0., 0., 1., 1., 1., 1., 0.],
|
|
[0., 0., 0., 0., 0., 0., 0., 0.]])
|
|
self.assertTrue((circ == circ_ex).all())
|