From 90e8ba379bf559531a73df7ff6e23107efc9360e Mon Sep 17 00:00:00 2001 From: ernestoarbitrio Date: Wed, 30 Sep 2020 15:13:12 +0200 Subject: [PATCH] new tests in benchmarks --- tests/benchmarks/test_benchmarks.py | 23 ++++++++++++++++++++++- 1 file changed, 22 insertions(+), 1 deletion(-) diff --git a/tests/benchmarks/test_benchmarks.py b/tests/benchmarks/test_benchmarks.py index 47ad10a3c..972c2165d 100644 --- a/tests/benchmarks/test_benchmarks.py +++ b/tests/benchmarks/test_benchmarks.py @@ -1,10 +1,16 @@ # encoding: utf-8 +import pytest + import numpy as np from histolab.filters.util import mask_difference, mask_percent +from histolab.util import np_to_pil, apply_mask_image + +from ..fixtures import NPY +from ..unitutil import PILIMG, NpArrayMock -class TestDescribeBenchmarksFilterUnit: +class TestDescribeBenchmarksFilterUtil: def test_mask_difference(self, benchmark): minuend = np.random.choice(a=[False, True], size=(5000, 4000)) subtrahend = np.random.choice(a=[False, True], size=(5000, 4000)) @@ -13,3 +19,18 @@ def test_mask_difference(self, benchmark): def test_mask_percent(self, benchmark): mask_array = np.random.choice(a=[False, True], size=(10000, 4000)) benchmark.pedantic(mask_percent, args=[mask_array], iterations=100, rounds=50) + + +class TestDescribeBenchmarksUtil: + def test_apply_mask_image(self, benchmark): + image = PILIMG.RGBA_COLOR_500X500_155_249_240 + mask = NpArrayMock.ONES_500X500X4_BOOL + benchmark.pedantic( + apply_mask_image, args=[image, mask], iterations=100, rounds=50 + ) + + @pytest.mark.parametrize( + "image", (NPY.NP_TO_PIL_RGBA, NPY.NP_TO_PIL_L, NPY.NP_TO_PIL_LA) + ) + def test_np_to_pil(self, image, benchmark): + benchmark.pedantic(np_to_pil, args=[image], iterations=500, rounds=250)