8000 Cleanup augmentables.utils tests by aleju · Pull Request #354 · aleju/imgaug · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Cleanup augmentables.utils tests #354

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Jul 16, 2019
Merged

Cleanup augmentables.utils tests #354

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
8000
Diff view
105 changes: 89 additions & 16 deletions test/augmentables/test_utils.py
View file
Open in desktop
8000 < 8000 /tr> 54F5
Original file line number Diff line number Diff line change
Expand Up @@ -16,18 +16,14 @@
matplotlib.use('Agg') # fix execution of tests involving matplotlib on travis
import numpy as np

from imgaug.testutils import reseed
from imgaug.augmentables.utils import (
interpolate_points, interpolate_point_pair,
interpolate_points_by_max_distance
)


class TestUtils(unittest.TestCase):
def setUp(self):
reseed()

def test_interpolate_point_pair(self):
class Test_interpolate_point_pair(unittest.TestCase):
def test_1_step(self):
point_a = (0, 0)
point_b = (1, 2)
inter = interpolate_point_pair(point_a, point_b, 1)
Expand All @@ -38,6 +34,9 @@ def test_interpolate_point_pair(self):
])
)

def test_2_steps(self):
point_a = (0, 0)
point_b = (1, 2)
inter = interpolate_point_pair(point_a, point_b, 2)
assert np.allclose(
inter,
Expand All @@ -47,16 +46,22 @@ def test_interpolate_point_pair(self):
])
)

def test_0_steps(self):
point_a = (0, 0)
point_b = (1, 2)
inter = interpolate_point_pair(point_a, point_b, 0)
assert len(inter) == 0

def test_interpolate_points(self):
# 2 points

class Test_interpolate_points(unittest.TestCase):
def test_2_points_0_steps(self):
points = [
(0, 0),
(1, 2)
]

inter = interpolate_points(points, 0)

assert np.allclose(
inter,
np.float32([
Expand All @@ -65,7 +70,14 @@ def test_interpolate_points(self):
])
)
def test_2_points_1_step(self):
points = [
(0, 0),
(1, 2)
]

inter = interpolate_points(points, 1)

assert np.allclose(
inter,
np.float32([
Expand All @@ -76,7 +88,14 @@ def test_interpolate_points(self):
])
)

def test_2_points_1_step_not_closed(self):
points = [
(0, 0),
(1, 2)
]

inter = interpolate_points(points, 1, closed=False)

assert np.allclose(
inter,
np.float32([
Expand All @@ -86,14 +105,15 @@ def test_interpolate_points(self):
])
)

# 3 points
def test_3_points_0_steps(self):
points = [
(0, 0),
(1, 2),
(0.5, 3)
]

inter = interpolate_points(points, 0)

assert np.allclose(
inter,
np.float32([
Expand All @@ -103,7 +123,15 @@ def test_interpolate_points(self):
])
)

def test_3_points_1_step(self):
points = [
(0, 0),
(1, 2),
(0.5, 3)
]

inter = interpolate_points(points, 1)

assert np.allclose(
inter,
np.float32([
Expand All @@ -116,7 +144,15 @@ def test_interpolate_points(self):
])
)

def test_3_points_1_step_not_closed(self):
points = [
(0, 0),
(1, 2),
(0.5, 3)
]

inter = interpolate_points(points, 1, closed=False)

assert np.allclose(
inter,
np.float32([
Expand All @@ -128,41 +164,60 @@ def test_interpolate_points(self):
])
)

# 0 points
def test_0_points_1_step(self):
points = []

inter = interpolate_points(points, 1)

assert len(inter) == 0

# 1 point
def test_1_point_0_steps(self):
points = [(0, 0)]

inter = interpolate_points(points, 0)

assert np.allclose(
inter,
np.float32([
[0, 0]
])
)

def test_1_point_1_step(self):
points = [(0, 0)]

inter = interpolate_points(points, 1)

assert np.allclose(
inter,
np.float32([
[0, 0]
])
)

def test_interpolate_points_by_max_distance(self):
# 2 points

class Test_interpolate_points_by_max_distance(unittest.TestCase):
def test_2_points_dist_10000(self):
points = [
(0, 0),
(0, 2)
]

inter = interpolate_points_by_max_distance(points, 10000)

assert np.allclose(
inter,
points
)

def test_2_points_dist_1(self):
points = [
(0, 0),
(0, 2)
]

inter = interpolate_points_by_max_distance(points, 1.0)

assert np.allclose(
inter,
np.float32([
Expand All @@ -173,7 +228,14 @@ def test_interpolate_points_by_max_distance(self):
])
)

def test_2_points_dist_1_not_closed(self):
points = [
(0, 0),
(0, 2)
]

inter = interpolate_points_by_max_distance(points, 1.0, closed=False)

assert np.allclose(
inter,
np.float32([
Expand All @@ -183,7 +245,7 @@ def test_interpolate_points_by_max_distance(self):
])
)

# 3 points
def test_3_points_dist_1(self):
points = [
(0, 0),
(0, 2),
Expand All @@ -203,7 +265,15 @@ def test_interpolate_points_by_max_distance(self):
])
)

def test_3_points_dist_1_not_closed(self):
points = [
(0, 0),
(0, 2),
(2, 0)
]

inter = interpolate_points_by_max_distance(points, 1.0, closed=False)

assert np.allclose(
inter,
np.float32([
Expand All @@ -215,15 +285,18 @@ def test_interpolate_points_by_max_distance(self):
])
)

# 0 points
def test_0_points_dist_1(self):
points = []

inter = interpolate_points_by_max_distance(points, 1.0)

assert len(inter) == 0

# 1 points
def test_1_point_dist_1(self):
points = [(0, 0)]

inter = interpolate_points_by_max_distance(points, 1.0)

assert np.allclose(
inter,
np.float32([
Expand Down
0