Skip to content
Snippets Groups Projects
Commit eacfa301 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[tests] Comment out outdated test sections

parent b7dac705
No related branches found
No related tags found
1 merge request!6Making use of LightningDataModule and simplification of data loading
Pipeline #76711 canceled
......@@ -6,10 +6,7 @@
import numpy
import PIL.Image
from ptbench.data.image_utils import ( # SingleAutoLevel16to8,
RemoveBlackBorders,
load_pil,
)
from ptbench.data.image_utils import RemoveBlackBorders
def test_remove_black_borders(datadir):
......@@ -32,23 +29,23 @@ def test_remove_black_borders(datadir):
numpy.testing.assert_array_equal(raw_without_black_border, raw_rbb_removed)
def test_load_pil_16bit(datadir):
# If the ratio is higher 0.5, image is probably clipped
Level16to8 = SingleAutoLevel16to8()
data_file = str(datadir / "16bits.png")
image = numpy.array(Level16to8(load_pil(data_file)))
count_pixels = numpy.count_nonzero(image)
count_max_value = numpy.count_nonzero(image == image.max())
assert count_max_value / count_pixels < 0.5
# It should not do anything to an image already in 8 bits
data_file = str(datadir / "raw_without_black_border.png")
img_loaded = load_pil(data_file)
original_8bits = numpy.array(img_loaded)
leveled_8bits = numpy.array(Level16to8(img_loaded))
numpy.testing.assert_array_equal(original_8bits, leveled_8bits)
# def test_load_pil_16bit(datadir):
# # If the ratio is higher 0.5, image is probably clipped
# Level16to8 = SingleAutoLevel16to8()
#
# data_file = str(datadir / "16bits.png")
# image = numpy.array(Level16to8(load_pil(data_file)))
#
# count_pixels = numpy.count_nonzero(image)
# count_max_value = numpy.count_nonzero(image == image.max())
#
# assert count_max_value / count_pixels < 0.5
#
# # It should not do anything to an image already in 8 bits
# data_file = str(datadir / "raw_without_black_border.png")
# img_loaded = load_pil(data_file)
#
# original_8bits = numpy.array(img_loaded)
# leveled_8bits = numpy.array(Level16to8(img_loaded))
#
# numpy.testing.assert_array_equal(original_8bits, leveled_8bits)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment