def test_init(): sdict = SortedDict() sdict._check() sdict = SortedDict(load=17) sdict._check() sdict = SortedDict((val, -val) for val in range(10000)) sdict._check() assert all(key == -val for key, val in sdict.iteritems()) sdict.clear() sdict._check() assert len(sdict) == 0 sdict = SortedDict.fromkeys(range(1000), None) assert all(sdict[key] == None for key in range(1000))
] df3['rebal_cnt_8'] = [ data['fromCNT'] if data is not None else 0 for data in df2[8] ] df3['rebal_cnt_9'] = [ data['fromCNT'] if data is not None else 0 for data in df2[9] ] k_means_1 = cluster.KMeans(n_clusters=4, init='k-means++', random_state=5000) data = df3[[ 'from_cnt_7', 'from_cnt_8', 'from_cnt_9', 'rebal_cnt_7', 'rebal_cnt_8', 'rebal_cnt_9' ]] k_means_1.fit(data) cluster1 = Counter(k_means_1.labels_) cluster1_sorted = SortedDict() cluster1_sorted = cluster1_sorted.fromkeys(cluster1.keys(), cluster1.values) # Kmean 2 k_means_2 = cluster.KMeans(n_clusters=5, init='k-means++', random_state=5000) k_means_2.fit(data) cluster2 = Counter(k_means_2.labels_) # Kmean 3 k_means_3 = cluster.KMeans(n_clusters=6, init='k-means++', random_state=5000) k_means_3.fit(data) cluster3 = Counter(k_means_3.labels_) # DBscan 1 DBscan_1 = cluster.DBSCAN(eps=20, min_samples=2).fit(data) cluster4 = Counter(DBscan_1.labels_) cluster4.pop(-1) # DBscan 2 DBscan_2 = cluster.DBSCAN(eps=15, min_samples=3).fit(data)
def test_fromkeys(): mapping = [(val, pos) for pos, val in enumerate(string.ascii_lowercase)] temp = SortedDict.fromkeys(mapping, 1) assert all(temp[key] == 1 for key in temp)
def fromkeys(cls, seq, value=None): d = DotMap() d._map = SortedDict.fromkeys(seq, value) return d