def test_getCrowdHierarchy(self): self.assertEqual( { 'cluster_6': 'cluster_8', 'cluster_5': 'cluster_8', 'cluster_2': 'cluster_5', 'cluster_3': 'cluster_6', 'cluster_8': 'cluster_10' }, AnalyzeData.getCrowdHierarchy('cluster_6')) self.assertEqual( { 'cluster_6': 'cluster_8', 'cluster_5': 'cluster_8', 'cluster_2': 'cluster_5', 'cluster_3': 'cluster_6', 'cluster_8': 'cluster_10' }, AnalyzeData.getCrowdHierarchy('cluster_10'))
def setUp(self): AnalyzeData.crowdMap, AnalyzeData.clusterIdToCrowdIdMap, AnalyzeData.crowdIdToClusterIdMap = {}, {}, {} self.clusterMaps = { test_time: [ {'clusterId': 'cluster_4', 'lastStreamAddedTime':getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_1'], 'streams': [], 'dimensions': {}}, {'clusterId': 'cluster_5', 'lastStreamAddedTime':getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_2'], 'streams': [], 'dimensions': {}}, {'clusterId': 'cluster_6', 'lastStreamAddedTime':getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_3'], 'streams': [], 'dimensions': {}}, ], test_time+timedelta(seconds=30*60): [ {'clusterId': 'cluster_7', 'lastStreamAddedTime':getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_4'], 'streams': [], 'dimensions': {}}, {'clusterId': 'cluster_8', 'lastStreamAddedTime':getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_5','cluster_6'], 'streams': [], 'dimensions': {}}, ], test_time+2*timedelta(seconds=30*60): [ {'clusterId': 'cluster_9', 'lastStreamAddedTime':getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_7'], 'streams': [], 'dimensions': {}}, {'clusterId': 'cluster_10', 'lastStreamAddedTime':getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_8'], 'streams': [], 'dimensions': {}}, ] } AnalyzeData.constructCrowdDataStructures(self.dataIterator)
def setUp(self): AnalyzeData.crowdMap, AnalyzeData.clusterIdToCrowdIdMap, AnalyzeData.crowdIdToClusterIdMap = {}, {}, {} self.clusterMaps = { test_time: [ { 'clusterId': 'cluster_4', 'lastStreamAddedTime': getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_1'], 'streams': [], 'dimensions': {} }, { 'clusterId': 'cluster_5', 'lastStreamAddedTime': getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_2'], 'streams': [], 'dimensions': {} }, { 'clusterId': 'cluster_6', 'lastStreamAddedTime': getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_3'], 'streams': [], 'dimensions': {} }, ], test_time + timedelta(seconds=30 * 60): [ { 'clusterId': 'cluster_7', 'lastStreamAddedTime': getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_4'], 'streams': [], 'dimensions': {} }, { 'clusterId': 'cluster_8', 'lastStreamAddedTime': getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_5', 'cluster_6'], 'streams': [], 'dimensions': {} }, ], test_time + 2 * timedelta(seconds=30 * 60): [ { 'clusterId': 'cluster_9', 'lastStreamAddedTime': getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_7'], 'streams': [], 'dimensions': {} }, { 'clusterId': 'cluster_10', 'lastStreamAddedTime': getStringRepresentationForTweetTimestamp(test_time), 'mergedClustersList': ['cluster_8'], 'streams': [], 'dimensions': {} }, ] } AnalyzeData.constructCrowdDataStructures(self.dataIterator)
def test_getCrowdHierarchy(self): self.assertEqual({'cluster_6': 'cluster_8', 'cluster_5': 'cluster_8', 'cluster_2': 'cluster_5', 'cluster_3': 'cluster_6', 'cluster_8': 'cluster_10'}, AnalyzeData.getCrowdHierarchy('cluster_6')) self.assertEqual({'cluster_6': 'cluster_8', 'cluster_5': 'cluster_8', 'cluster_2': 'cluster_5', 'cluster_3': 'cluster_6', 'cluster_8': 'cluster_10'}, AnalyzeData.getCrowdHierarchy('cluster_10'))