"""! @brief Unit-tests for OPTICS algorithm. @authors Andrei Novikov (pyclustering@yandex.ru) @date 2014-2020 @copyright BSD-3-Clause """ import unittest # Generate images without having a window appear. import matplotlib matplotlib.use('Agg') from pyclustering.cluster.tests.optics_templates import OpticsTestTemplates from pyclustering.cluster.optics import optics, ordering_analyser, ordering_visualizer from pyclustering.utils import read_sample from pyclustering.samples.definitions import SIMPLE_SAMPLES, FCPS_SAMPLES class OpticsUnitTest(unittest.TestCase): def testClusteringSampleSimple1(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE1, 0.4, 2, None, [5, 5], False) def testClusteringSampleSimple1DistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE1, 0.4, 2, None, [5, 5], False) def testClusteringSampleSimple2(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE2, 1, 2, None, [5, 8, 10], False) def testClusteringSampleSimple2DistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE2, 1, 2, None, [5, 8, 10], False) def testClusteringSampleSimple3(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE3, 0.7, 3, None, [10, 10, 10, 30], False) def testClusteringSampleSimple3DistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE3, 0.7, 3, None, [10, 10, 10, 30], False) def testClusteringSampleSimple4(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE4, 0.7, 3, None, [15, 15, 15, 15, 15], False) def testClusteringSampleSimple4DistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE4, 0.7, 3, None, [15, 15, 15, 15, 15], False) def testClusteringSampleSimple5(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE5, 0.7, 3, None, [15, 15, 15, 15], False) def testClusteringSampleSimple5DistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE5, 0.7, 3, None, [15, 15, 15, 15], False) def testClusteringSampleSimple6(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE6, 1.0, 3, None, [20, 21], False) def testClusteringSampleSimple6DistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE6, 1.0, 3, None, [20, 21], False) def testClusteringThreeDimensionalSimple11(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE11, 2.0, 2, None, [10, 10], False) OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE11, 3.0, 2, None, [10, 10], False) def testClusteringTheSameData1(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE9, 1.0, 3, None, [10, 20], False) def testClusteringTheSameData1DistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE9, 1.0, 3, None, [10, 20], False) def testClusteringTheSameData2(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE12, 1.0, 2, None, [5, 5, 5], False) def testClusteringTheSameData2DistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE12, 1.0, 2, None, [5, 5, 5], False) def testClusteringTheSameData2OneCluster(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE12, 20.0, 2, None, [15], False) def testClusteringTheSameData2OneClusterDistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE12, 20.0, 2, None, [15], False) def testClusteringHepta(self): OpticsTestTemplates.templateClusteringResults(FCPS_SAMPLES.SAMPLE_HEPTA, 1, 3, None, [30, 30, 30, 30, 30, 30, 32], False) def testClusteringHeptaDistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(FCPS_SAMPLES.SAMPLE_HEPTA, 1, 3, None, [30, 30, 30, 30, 30, 30, 32], False) def testClusteringOneDimensionDataSampleSimple7(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE7, 2.0, 2, None, [10, 10], False) def testClusteringOneDimensionDataSampleSimple7DistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE7, 2.0, 2, None, [10, 10], False) def testClusteringOneDimensionDataSampleSimple9(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE9, 3.0, 3, None, [10, 20], False) def testClusteringOneDimensionDataSampleSimple9DistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE9, 3.0, 3, None, [10, 20], False) def testClusteringSampleSimple2RadiusGreater(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE2, 5.0, 2, 3, [5, 8, 10], False) def testClusteringSampleSimple2RadiusGreaterDistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE2, 5.0, 2, 3, [5, 8, 10], False) def testClusteringSampleSimple3RadiusGreater(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE3, 5.0, 3, 4, [10, 10, 10, 30], False) def testClusteringSampleSimple3RadiusGreaterDistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE3, 5.0, 3, 4, [10, 10, 10, 30], False) def testClusteringSampleSimple4RadiusGreater(self): OpticsTestTemplates.templateClusteringResults(SIMPLE_SAMPLES.SAMPLE_SIMPLE4, 6.0, 3, 5, [15, 15, 15, 15, 15], False) def testClusteringSampleSimple4RadiusGreaterDistanceMatrix(self): OpticsTestTemplates.templateClusteringResultsDistanceMatrix(SIMPLE_SAMPLES.SAMPLE_SIMPLE4, 6.0, 3, 5, [15, 15, 15, 15, 15], False) def testClusteringLsunRadiusGreater(self): # Two points are noise after recalculation OpticsTestTemplates.templateClusteringResults(FCPS_SAMPLES.SAMPLE_LSUN, 1.0, 3, 3, [99, 100, 202], False) def testClusteringOrderVisualizer(self): sample = read_sample(SIMPLE_SAMPLES.SAMPLE_SIMPLE4) optics_instance = optics(sample, 6.0, 3, 5) optics_instance.process() analyser = ordering_analyser(optics_instance.get_ordering()) ordering_visualizer.show_ordering_diagram(analyser, 5) def testClusterOrderingOneClusterExtraction(self): analyser = ordering_analyser([5.0, 5.0, 5.0, 5.0, 5.0, 5.0]) amount_clusters, borders = analyser.extract_cluster_amount(6.5) assert 1 == amount_clusters assert 0 == len(borders) amount_clusters, borders = analyser.extract_cluster_amount(4.5) assert 0 == amount_clusters assert 0 == len(borders) def testImpossibleClusterOrderingAllocationHomogeneous(self): analyser = ordering_analyser([5.0, 5.0, 5.0, 5.0, 5.0, 5.0, 5.0, 5.0, 5.0, 5.0]) amount_clusters, borders = analyser.calculate_connvectivity_radius(2) assert None == amount_clusters; assert 0 == len(borders); def testImpossibleClusterOrderingAllocationGeterogeneous(self): analyser = ordering_analyser([5.0, 5.0, 5.0, 5.0, 6.0, 8.0, 6.0, 5.0, 5.0, 5.0]) amount_clusters, borders = analyser.calculate_connvectivity_radius(3) assert None == amount_clusters assert 0 == len(borders) def test_incorrect_data(self): self.assertRaises(ValueError, optics, [], 0.1, 1) def test_incorrect_eps(self): self.assertRaises(ValueError, optics, [[0], [1], [2]], -1.0, 1) def test_incorrect_minpts(self): self.assertRaises(ValueError, optics, [[0], [1], [2]], 0.5, -1) def test_incorrect_amount_clusters(self): self.assertRaises(ValueError, optics, [[0], [1], [2]], 0.5, 1, amount_clusters=-1)