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'))
