def test_dbscan_clustering(): # Test dbscan import shutil # Copy alist to the output directory so that the DBSCAN results end up in the right place. alist_data = storm_analysis.getData("test/data/test_clustering_list.bin") alist_output = storm_analysis.getPathOutputTest( "test_clustering_alist.bin") shutil.copyfile(alist_data, alist_output) from storm_analysis.dbscan.dbscan_analysis import dbscanAnalysis dbscanAnalysis(alist_output, 0) # Verify number of clusters found. stats_file = storm_analysis.getPathOutputTest( "test_clustering_aclusters_stats.txt") n_clusters = len(open(stats_file).readlines()) if (n_clusters != 99): raise Exception( "DBSCAN did not identify the expected number of clusters.") # Make pictures. clist_name = storm_analysis.getPathOutputTest( "test_clustering_aclusters_size_list.bin") image_name = storm_analysis.getPathOutputTest("test_clustering_db") from storm_analysis.dbscan.cluster_images import clusterImages clusterImages(clist_name, "DBSCAN Clustering", 50, 20, image_name, [256, 256])
def test_voronoi_clustering(): # Test voronoi alist_name = storm_analysis.getData("test/data/test_clustering_list.bin") output_dir = storm_analysis.getPathOutputTest("./") from storm_analysis.voronoi.voronoi_analysis import voronoiAnalysis voronoiAnalysis(alist_name, 0.1, output_dir) # Verify number of clusters found. stats_file = storm_analysis.getPathOutputTest( "test_clustering_srt_stats.txt") n_clusters = len(open(stats_file).readlines()) if (n_clusters != 100): raise Exception( "Voronoi did not identify the expected number of clusters.") # Make pictures. clist_name = storm_analysis.getPathOutputTest( "test_clustering_srt_size_list.bin") image_name = storm_analysis.getPathOutputTest("test_clustering_vr") from storm_analysis.dbscan.cluster_images import clusterImages clusterImages(clist_name, "Voronoi Clustering", 50, 20, image_name, [256, 256])
def test_voronoi_clustering(): # Test voronoi alist_name = storm_analysis.getData("test/data/test_drift_alist.bin") output_dir = storm_analysis.getPathOutputTest("./") from storm_analysis.voronoi.voronoi_analysis import voronoiAnalysis voronoiAnalysis(alist_name, 1.25, output_dir) clist_name = storm_analysis.getPathOutputTest("test_drift_asrt_size_list.bin") image_name = storm_analysis.getPathOutputTest("test_drift_vr") from storm_analysis.dbscan.cluster_images import clusterImages clusterImages(clist_name, "Voronoi Clustering", 50, 20, image_name, [256, 256])
def test_dbscan_clustering(): # Test dbscan import shutil # Copy alist to the output directory so that the DBSCAN results end up in the right place. alist_data = storm_analysis.getData("test/data/test_drift_alist.bin") alist_output = storm_analysis.getPathOutputTest("test_drift_alist.bin") shutil.copyfile(alist_data, alist_output) from storm_analysis.dbscan.dbscan_analysis import dbscanAnalysis dbscanAnalysis(alist_output, 0) clist_name = storm_analysis.getPathOutputTest("test_drift_aclusters_size_list.bin") image_name = storm_analysis.getPathOutputTest("test_drift_db") from storm_analysis.dbscan.cluster_images import clusterImages clusterImages(clist_name, "DBSCAN Clustering", 50, 20, image_name, [256, 256])