tests.color_segmentation package

Submodules

tests.color_segmentation.manual_testing_color_seg module

visualize_clusters(small_image: array, clusters: array)

Visualize the color clusters of the resized image

Parameters:

• small_image – the resized image

• clusters – the list of clusters obtained

Returns:

None

tests.color_segmentation.test_color_seg module

class TestColorSegmentation

Bases: object

BITMASK_PATH: str = 'tests/resources/color_segmentation/data_source/bitmask.png'

BW_IMAGE_PATH: str = 'tests/resources/color_segmentation/black_and_white_image.png'

CUSTOM_CONFIG_PATH: str = 'tests/resources/configs/color-segmentation.yml'

DATA_CUBE_PATH: str = 'tests/resources/color_segmentation/data_source/test_cube.dms'

DATA_SOURCE = 'data_source'

IMAGE_NAME = 'RGB'

PATH_DATA_SOURCE: str = 'tests/resources/color_segmentation/data_source'

REG_TEST_IMAGE_PATH: str = 'tests/resources/color_segmentation/data_source/registered_test_image.png'

TEST_IMAGE_PATH: str = 'tests/resources/color_segmentation/data_source/test_image_cs.png'

WRONG_CONFIG_PATH: str = 'tests/resources/configs/random.yml'

elem_threshold: float = 0.1

k: int = 2

num_attempts: int = 10

setup_get_elemental_clusters(missing_param: str)

test_combined_bitmasks()

test_combined_empty_bitmasks()

test_convert_to_hex()

test_get_clusters_using_k_means_colors(caplog)

test_get_clusters_using_k_means_colors_register_fail(caplog)

test_get_elem_clusters_using_k_means()

test_get_elem_clusters_using_k_means_elemental_not_found(caplog)

test_get_elem_clusters_using_k_means_empty_mask()

test_get_elem_clusters_using_k_means_register_fail(caplog)

test_get_path_to_cs_folder()

test_get_path_to_cs_folder_wrong_datasource(caplog)

test_image_to_rgb_and_lab()

test_merge_similar_colors(caplog)

test_merge_similar_colors_empty(caplog)

test_rgb_to_lab()

test_save_bitmask()

test_save_bitmask_invalid_bitmask(caplog)

test_save_bitmask_wrong_path(caplog)

empty_array(array: list[ndarray] | list[list[ndarray]])

Module contents