def test_edge_detection(self): # test canny method samples = IA.edge_detection(test_image[12:105, 17:148], 24) assert isinstance(samples, list), 'the output should be a list' assert len(samples) == 24, \ 'number of samples found exceeds the expected one' # test sobel method samples = IA.edge_detection(test_image[12:105, 17:148], 24, method='sobel') assert isinstance(samples, list), 'the output should be a list' assert len(samples) == 24, \ 'number of samples found exceeds the expected one' return
def test_sort_samples(self): samples = IA.edge_detection(test_image[12:105, 17:148], 24) sample_location = IA.sample_locations(samples, 24) sorted = IA.sort_samples(sample_location, 6, 4) assert len(sorted) == 24, 'resulting dataframe is not the ' +\ 'correct length' assert len(sorted.columns) == 2, 'the dataframe contains extra ' +\ 'columns. Expected number was 2, found {}'.format( len(sorted.columns)) return
import unittest import numpy as np from phasIR import image_analysis as IA from unittest.mock import patch image_file = './doc/data/empty_plates_images/24_conical_empty_plate.png' video_file = './phasIR/test/data/empty_plate_well_position.HDF5' test_image = IA.input_file(image_file) samples = IA.edge_detection(test_image[12:105, 17:148], 24) sample_location = IA.sample_locations(samples, 24) sorted = IA.sort_samples(sample_location, 6, 4) mock_coords = [[1, 13], [24, 25]] class TestSimulationTools(unittest.TestCase): def test_input_file(self): '''Test for function which loads the input file''' frames = IA.input_file(image_file) assert isinstance(frames, np.ndarray), 'Output is not an array' frames = IA.input_file(video_file) assert isinstance(frames, list), 'Output is not an array' return def test_flip_frame(self): frames = IA.input_file(video_file) flipped_frames = IA.flip_frame(frames) assert isinstance(flipped_frames, list), 'Output is not an array' return