Skip to content

Instantly share code, notes, and snippets.

@prashantdce19
Created July 25, 2014 10:16
Show Gist options
  • Save prashantdce19/b680b0a8667a11731899 to your computer and use it in GitHub Desktop.
Save prashantdce19/b680b0a8667a11731899 to your computer and use it in GitHub Desktop.
Unitests for similar_users_update_with_user_network.py
import unittest
from similar_users_update_with_user_network import *
from bson.objectid import ObjectId
class convflowDB(unittest.TestCase):
def test_count_keywords(self):
self.assertEqual(count_keywords([],[]),0)
self.assertEqual(count_keywords(['ab'],['ba']),0)
self.assertEqual(count_keywords(['ab'],['ab']),1)
self.assertEqual(count_keywords(['one','two'],['two','one']),2)
self.assertEqual(count_keywords(['one','two','three'],[]),0)
self.assertEqual(count_keywords([],['one','two','three']),0)
def test_find_similiar_problems(self):
self.assertEqual(find_similiar_problems([]),[])
self.assertEqual(find_similiar_problems(['innovative']),[(ObjectId('53d0b0bcd8a292144ce5c1c6'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e3'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e4'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e9'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ee'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f6'), 1), (ObjectId('53d0b0bcd8a292144ce5c19e'), 1), (ObjectId('53d0b0bcd8a292144ce5c1a0'), 1), (ObjectId('53d0b0bcd8a292144ce5c1a3'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ae'), 1), (ObjectId('53d0b0bcd8a292144ce5c1af'), 1), (ObjectId('53d0b0bcd8a292144ce5c1b2'), 1), (ObjectId('53d0b0bcd8a292144ce5c1bc'), 1), (ObjectId('53d0b0bcd8a292144ce5c1c2'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ca'), 1), (ObjectId('53d0b0bcd8a292144ce5c1d5'), 1), (ObjectId('53d0b0bcd8a292144ce5c1da'), 1)])
self.assertEqual(find_similiar_problems(['Noinnovative']),[])
self.assertEqual(find_similiar_problems(['innovative','creative','innovation','india']),[(ObjectId('53d0b0bcd8a292144ce5c1e9'), 2), (ObjectId('53d0b0bcd8a292144ce5c1ae'), 2), (ObjectId('53d0b0bcd8a292144ce5c1c1'), 1), (ObjectId('53d0b0bcd8a292144ce5c1c6'), 1), (ObjectId('53d0b0bcd8a292144ce5c1dc'), 1), (ObjectId('53d0b0bcd8a292144ce5c1de'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e0'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e1'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e3'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e4'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e7'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ee'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ef'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f1'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f2'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f5'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f6'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f8'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f9'), 1), (ObjectId('53d0b0bcd8a292144ce5c1fa'), 1), (ObjectId('53d0b0bcd8a292144ce5c1fd'), 1), (ObjectId('53d0b0bcd8a292144ce5c19d'), 1), (ObjectId('53d0b0bcd8a292144ce5c19e'), 1), (ObjectId('53d0b0bcd8a292144ce5c1a0'), 1), (ObjectId('53d0b0bcd8a292144ce5c1a3'), 1), (ObjectId('53d0b0bcd8a292144ce5c1a5'), 1), (ObjectId('53d0b0bcd8a292144ce5c1a9'), 1), (ObjectId('53d0b0bcd8a292144ce5c1aa'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ac'), 1), (ObjectId('53d0b0bcd8a292144ce5c1af'), 1), (ObjectId('53d0b0bcd8a292144ce5c1b2'), 1), (ObjectId('53d0b0bcd8a292144ce5c1b4'), 1), (ObjectId('53d0b0bcd8a292144ce5c1b5'), 1), (ObjectId('53d0b0bcd8a292144ce5c1b9'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ba'), 1), (ObjectId('53d0b0bcd8a292144ce5c1bb'), 1), (ObjectId('53d0b0bcd8a292144ce5c1bc'), 1), (ObjectId('53d0b0bcd8a292144ce5c1c2'), 1), (ObjectId('53d0b0bcd8a292144ce5c1c5'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ca'), 1), (ObjectId('53d0b0bcd8a292144ce5c1d4'), 1), (ObjectId('53d0b0bcd8a292144ce5c1d5'), 1), (ObjectId('53d0b0bcd8a292144ce5c1d6'), 1), (ObjectId('53d0b0bcd8a292144ce5c1da'), 1), (ObjectId('53d0b0bcd8a292144ce5c1db'), 1)])
def test_network_participants(self):
self.assertEqual(network_participants(ObjectId()),[])
self.assertEqual(network_participants(ObjectId('53d0b0bcd8a292144ce5c1c1')),[])
self.assertEqual(network_participants(ObjectId('53d0b0bbd8a292144ce5c198')),[(ObjectId('53d0b0bbd8a292144ce5c181'), 9), (ObjectId('53d0b0bbd8a292144ce5c182'), 14), (ObjectId('53d0b0bbd8a292144ce5c190'), 9), (ObjectId('53d0b0bbd8a292144ce5c16b'), 16), (ObjectId('53d0b0bbd8a292144ce5c17d'), 12), (ObjectId('53d0b0bbd8a292144ce5c14b'), 10), (ObjectId('53d0b0bbd8a292144ce5c194'), 17), (ObjectId('53d0b0bbd8a292144ce5c167'), 11), (ObjectId('53d0b0bbd8a292144ce5c176'), 14), (ObjectId('53d0b0bbd8a292144ce5c17b'), 18), (ObjectId('53d0b0bbd8a292144ce5c17f'), 15), (ObjectId('53d0b0bbd8a292144ce5c193'), 10), (ObjectId('53d0b0bbd8a292144ce5c160'), 8), (ObjectId('53d0b0bbd8a292144ce5c174'), 9), (ObjectId('53d0b0bbd8a292144ce5c17a'), 15), (ObjectId('53d0b0bbd8a292144ce5c179'), 9), (ObjectId('53d0b0bbd8a292144ce5c146'), 15), (ObjectId('53d0b0bbd8a292144ce5c15e'), 14), (ObjectId('53d0b0bbd8a292144ce5c16d'), 20), (ObjectId('53d0b0bbd8a292144ce5c13c'), 12), (ObjectId('53d0b0bbd8a292144ce5c173'), 12), (ObjectId('53d0b0bbd8a292144ce5c139'), 16), (ObjectId('53d0b0bbd8a292144ce5c170'), 13), (ObjectId('53d0b0bbd8a292144ce5c147'), 17), (ObjectId('53d0b0bbd8a292144ce5c148'), 11), (ObjectId('53d0b0bbd8a292144ce5c140'), 13), (ObjectId('53d0b0bbd8a292144ce5c157'), 15), (ObjectId('53d0b0bbd8a292144ce5c19a'), 13), (ObjectId('53d0b0bbd8a292144ce5c154'), 14), (ObjectId('53d0b0bbd8a292144ce5c15f'), 20), (ObjectId('53d0b0bbd8a292144ce5c15c'), 16), (ObjectId('53d0b0bbd8a292144ce5c172'), 17), (ObjectId('53d0b0bbd8a292144ce5c13b'), 15), (ObjectId('53d0b0bbd8a292144ce5c171'), 14), (ObjectId('53d0b0bbd8a292144ce5c13a'), 12), (ObjectId('53d0b0bbd8a292144ce5c14e'), 15), (ObjectId('53d0b0bbd8a292144ce5c184'), 13), (ObjectId('53d0b0bbd8a292144ce5c191'), 11), (ObjectId('53d0b0bbd8a292144ce5c169'), 10), (ObjectId('53d0b0bbd8a292144ce5c138'), 13), (ObjectId('53d0b0bbd8a292144ce5c16f'), 10), (ObjectId('53d0b0bbd8a292144ce5c15a'), 13), (ObjectId('53d0b0bbd8a292144ce5c143'), 13), (ObjectId('53d0b0bbd8a292144ce5c195'), 11), (ObjectId('53d0b0bbd8a292144ce5c14f'), 11), (ObjectId('53d0b0bbd8a292144ce5c14c'), 13), (ObjectId('53d0b0bbd8a292144ce5c13f'), 10), (ObjectId('53d0b0bbd8a292144ce5c156'), 10), (ObjectId('53d0b0bbd8a292144ce5c144'), 14), (ObjectId('53d0b0bbd8a292144ce5c15b'), 10), (ObjectId('53d0b0bbd8a292144ce5c178'), 18), (ObjectId('53d0b0bbd8a292144ce5c155'), 14), (ObjectId('53d0b0bbd8a292144ce5c13e'), 12), (ObjectId('53d0b0bbd8a292144ce5c15d'), 8), (ObjectId('53d0b0bbd8a292144ce5c177'), 12), (ObjectId('53d0b0bbd8a292144ce5c164'), 14), (ObjectId('53d0b0bbd8a292144ce5c13d'), 15), (ObjectId('53d0b0bbd8a292144ce5c187'), 18), (ObjectId('53d0b0bbd8a292144ce5c185'), 14), (ObjectId('53d0b0bbd8a292144ce5c14a'), 13), (ObjectId('53d0b0bbd8a292144ce5c18b'), 14), (ObjectId('53d0b0bbd8a292144ce5c188'), 15), (ObjectId('53d0b0bbd8a292144ce5c18c'), 11), (ObjectId('53d0b0bbd8a292144ce5c163'), 17), (ObjectId('53d0b0bbd8a292144ce5c17c'), 12), (ObjectId('53d0b0bbd8a292144ce5c159'), 11), (ObjectId('53d0b0bbd8a292144ce5c142'), 9), (ObjectId('53d0b0bbd8a292144ce5c162'), 20), (ObjectId('53d0b0bbd8a292144ce5c158'), 14), (ObjectId('53d0b0bbd8a292144ce5c141'), 8), (ObjectId('53d0b0bbd8a292144ce5c161'), 15), (ObjectId('53d0b0bbd8a292144ce5c16e'), 10), (ObjectId('53d0b0bbd8a292144ce5c16c'), 14), (ObjectId('53d0b0bbd8a292144ce5c199'), 15), (ObjectId('53d0b0bbd8a292144ce5c153'), 15), (ObjectId('53d0b0bbd8a292144ce5c180'), 15), (ObjectId('53d0b0bbd8a292144ce5c17e'), 13), (ObjectId('53d0b0bbd8a292144ce5c186'), 16), (ObjectId('53d0b0bbd8a292144ce5c145'), 11), (ObjectId('53d0b0bbd8a292144ce5c152'), 14), (ObjectId('53d0b0bbd8a292144ce5c165'), 15), (ObjectId('53d0b0bbd8a292144ce5c18d'), 15), (ObjectId('53d0b0bbd8a292144ce5c166'), 16), (ObjectId('53d0b0bbd8a292144ce5c18e'), 14), (ObjectId('53d0b0bbd8a292144ce5c183'), 12), (ObjectId('53d0b0bbd8a292144ce5c14d'), 11), (ObjectId('53d0b0bbd8a292144ce5c197'), 7), (ObjectId('53d0b0bbd8a292144ce5c151'), 6), (ObjectId('53d0b0bbd8a292144ce5c16a'), 12), (ObjectId('53d0b0bbd8a292144ce5c192'), 14), (ObjectId('53d0b0bbd8a292144ce5c168'), 13), (ObjectId('53d0b0bbd8a292144ce5c150'), 11), (ObjectId('53d0b0bbd8a292144ce5c196'), 12), (ObjectId('53d0b0bbd8a292144ce5c189'), 11), (ObjectId('53d0b0bbd8a292144ce5c149'), 12), (ObjectId('53d0b0bbd8a292144ce5c18a'), 9), (ObjectId('53d0b0bbd8a292144ce5c18f'), 17), (ObjectId('53d0b0bbd8a292144ce5c175'), 12)])
def test_user_activity_count(self):
userids = [i['_id'] for i in users.find()]
problemids = [i['_id'] for i in problems.find()]
self.assertEqual(user_activity_count([]),[])
self.assertEqual(user_activity_count(userids),[])
self.assertEqual(user_activity_count(problemids),[{u'count': 38, u'_id': ObjectId('53d0b0bbd8a292144ce5c174')}, {u'count': 33, u'_id': ObjectId('53d0b0bbd8a292144ce5c17a')}, {u'count': 30, u'_id': ObjectId('53d0b0bbd8a292144ce5c179')}, {u'count': 41, u'_id': ObjectId('53d0b0bbd8a292144ce5c14c')}, {u'count': 33, u'_id': ObjectId('53d0b0bbd8a292144ce5c177')}, {u'count': 43, u'_id': ObjectId('53d0b0bbd8a292144ce5c178')}, {u'count': 39, u'_id': ObjectId('53d0b0bbd8a292144ce5c17e')}, {u'count': 43, u'_id': ObjectId('53d0b0bbd8a292144ce5c146')}, {u'count': 45, u'_id': ObjectId('53d0b0bbd8a292144ce5c188')}, {u'count': 37, u'_id': ObjectId('53d0b0bbd8a292144ce5c192')}, {u'count': 43, u'_id': ObjectId('53d0b0bbd8a292144ce5c161')}, {u'count': 40, u'_id': ObjectId('53d0b0bbd8a292144ce5c166')}, {u'count': 41, u'_id': ObjectId('53d0b0bbd8a292144ce5c165')}, {u'count': 34, u'_id': ObjectId('53d0b0bbd8a292144ce5c17f')}, {u'count': 38, u'_id': ObjectId('53d0b0bbd8a292144ce5c147')}, {u'count': 40, u'_id': ObjectId('53d0b0bbd8a292144ce5c18b')}, {u'count': 38, u'_id': ObjectId('53d0b0bbd8a292144ce5c163')}, {u'count': 45, u'_id': ObjectId('53d0b0bbd8a292144ce5c14e')}, {u'count': 35, u'_id': ObjectId('53d0b0bbd8a292144ce5c144')}, {u'count': 31, u'_id': ObjectId('53d0b0bbd8a292144ce5c15b')}, {u'count': 45, u'_id': ObjectId('53d0b0bbd8a292144ce5c15f')}, {u'count': 36, u'_id': ObjectId('53d0b0bbd8a292144ce5c173')}, {u'count': 32, u'_id': ObjectId('53d0b0bbd8a292144ce5c13c')}, {u'count': 37, u'_id': ObjectId('53d0b0bbd8a292144ce5c151')}, {u'count': 34, u'_id': ObjectId('53d0b0bbd8a292144ce5c197')}, {u'count': 37, u'_id': ObjectId('53d0b0bbd8a292144ce5c13d')}, {u'count': 45, u'_id': ObjectId('53d0b0bbd8a292144ce5c164')}, {u'count': 37, u'_id': ObjectId('53d0b0bbd8a292144ce5c16a')}, {u'count': 32, u'_id': ObjectId('53d0b0bbd8a292144ce5c16e')}, {u'count': 38, u'_id': ObjectId('53d0b0bbd8a292144ce5c17b')}, {u'count': 30, u'_id': ObjectId('53d0b0bbd8a292144ce5c145')}, {u'count': 48, u'_id': ObjectId('53d0b0bbd8a292144ce5c199')}, {u'count': 37, u'_id': ObjectId('53d0b0bbd8a292144ce5c153')}, {u'count': 29, u'_id': ObjectId('53d0b0bbd8a292144ce5c13f')}, {u'count': 34, u'_id': ObjectId('53d0b0bbd8a292144ce5c156')}, {u'count': 42, u'_id': ObjectId('53d0b0bbd8a292144ce5c172')}, {u'count': 39, u'_id': ObjectId('53d0b0bbd8a292144ce5c13b')}, {u'count': 38, u'_id': ObjectId('53d0b0bbd8a292144ce5c155')}, {u'count': 45, u'_id': ObjectId('53d0b0bbd8a292144ce5c13e')}, {u'count': 45, u'_id': ObjectId('53d0b0bbd8a292144ce5c181')}, {u'count': 41, u'_id': ObjectId('53d0b0bbd8a292144ce5c180')}, {u'count': 36, u'_id': ObjectId('53d0b0bbd8a292144ce5c160')}, {u'count': 42, u'_id': ObjectId('53d0b0bbd8a292144ce5c16d')}, {u'count': 39, u'_id': ObjectId('53d0b0bbd8a292144ce5c194')}, {u'count': 39, u'_id': ObjectId('53d0b0bbd8a292144ce5c16c')}, {u'count': 40, u'_id': ObjectId('53d0b0bbd8a292144ce5c13a')}, {u'count': 39, u'_id': ObjectId('53d0b0bbd8a292144ce5c171')}, {u'count': 48, u'_id': ObjectId('53d0b0bbd8a292144ce5c157')}, {u'count': 41, u'_id': ObjectId('53d0b0bbd8a292144ce5c140')}, {u'count': 37, u'_id': ObjectId('53d0b0bbd8a292144ce5c142')}, {u'count': 34, u'_id': ObjectId('53d0b0bbd8a292144ce5c159')}, {u'count': 42, u'_id': ObjectId('53d0b0bbd8a292144ce5c170')}, {u'count': 42, u'_id': ObjectId('53d0b0bbd8a292144ce5c139')}, {u'count': 45, u'_id': ObjectId('53d0b0bbd8a292144ce5c182')}, {u'count': 40, u'_id': ObjectId('53d0b0bbd8a292144ce5c18d')}, {u'count': 44, u'_id': ObjectId('53d0b0bbd8a292144ce5c15a')}, {u'count': 39, u'_id': ObjectId('53d0b0bbd8a292144ce5c143')}, {u'count': 32, u'_id': ObjectId('53d0b0bbd8a292144ce5c14f')}, {u'count': 35, u'_id': ObjectId('53d0b0bbd8a292144ce5c195')}, {u'count': 36, u'_id': ObjectId('53d0b0bbd8a292144ce5c17c')}, {u'count': 49, u'_id': ObjectId('53d0b0bbd8a292144ce5c162')}, {u'count': 49, u'_id': ObjectId('53d0b0bbd8a292144ce5c18f')}, {u'count': 39, u'_id': ObjectId('53d0b0bbd8a292144ce5c167')}, {u'count': 38, u'_id': ObjectId('53d0b0bbd8a292144ce5c191')}, {u'count': 35, u'_id': ObjectId('53d0b0bbd8a292144ce5c169')}, {u'count': 33, u'_id': ObjectId('53d0b0bbd8a292144ce5c138')}, {u'count': 41, u'_id': ObjectId('53d0b0bbd8a292144ce5c16f')}, {u'count': 35, u'_id': ObjectId('53d0b0bbd8a292144ce5c14a')}, {u'count': 36, u'_id': ObjectId('53d0b0bbd8a292144ce5c14d')}, {u'count': 39, u'_id': ObjectId('53d0b0bbd8a292144ce5c183')}, {u'count': 43, u'_id': ObjectId('53d0b0bbd8a292144ce5c189')}, {u'count': 44, u'_id': ObjectId('53d0b0bbd8a292144ce5c148')}, {u'count': 41, u'_id': ObjectId('53d0b0bbd8a292144ce5c16b')}, {u'count': 38, u'_id': ObjectId('53d0b0bbd8a292144ce5c193')}, {u'count': 33, u'_id': ObjectId('53d0b0bbd8a292144ce5c198')}, {u'count': 38, u'_id': ObjectId('53d0b0bbd8a292144ce5c152')}, {u'count': 45, u'_id': ObjectId('53d0b0bbd8a292144ce5c150')}, {u'count': 37, u'_id': ObjectId('53d0b0bbd8a292144ce5c196')}, {u'count': 41, u'_id': ObjectId('53d0b0bbd8a292144ce5c18e')}, {u'count': 46, u'_id': ObjectId('53d0b0bbd8a292144ce5c15e')}, {u'count': 36, u'_id': ObjectId('53d0b0bbd8a292144ce5c15d')}, {u'count': 42, u'_id': ObjectId('53d0b0bbd8a292144ce5c168')}, {u'count': 38, u'_id': ObjectId('53d0b0bbd8a292144ce5c190')}, {u'count': 36, u'_id': ObjectId('53d0b0bbd8a292144ce5c15c')}, {u'count': 43, u'_id': ObjectId('53d0b0bbd8a292144ce5c187')}, {u'count': 48, u'_id': ObjectId('53d0b0bbd8a292144ce5c186')}, {u'count': 39, u'_id': ObjectId('53d0b0bbd8a292144ce5c149')}, {u'count': 43, u'_id': ObjectId('53d0b0bbd8a292144ce5c184')}, {u'count': 40, u'_id': ObjectId('53d0b0bbd8a292144ce5c18a')}, {u'count': 41, u'_id': ObjectId('53d0b0bbd8a292144ce5c141')}, {u'count': 41, u'_id': ObjectId('53d0b0bbd8a292144ce5c158')}, {u'count': 44, u'_id': ObjectId('53d0b0bbd8a292144ce5c154')}, {u'count': 40, u'_id': ObjectId('53d0b0bbd8a292144ce5c19a')}, {u'count': 41, u'_id': ObjectId('53d0b0bbd8a292144ce5c175')}, {u'count': 35, u'_id': ObjectId('53d0b0bbd8a292144ce5c14b')}, {u'count': 43, u'_id': ObjectId('53d0b0bbd8a292144ce5c18c')}, {u'count': 41, u'_id': ObjectId('53d0b0bbd8a292144ce5c176')}, {u'count': 34, u'_id': ObjectId('53d0b0bbd8a292144ce5c17d')}, {u'count': 36, u'_id': ObjectId('53d0b0bbd8a292144ce5c185')}])
def test_distribute(self):
smthng = [(ObjectId('53d0b0bcd8a292144ce5c1e9'), 2), (ObjectId('53d0b0bcd8a292144ce5c1ae'), 2), (ObjectId('53d0b0bcd8a292144ce5c1c1'), 1), (ObjectId('53d0b0bcd8a292144ce5c1c6'), 1), (ObjectId('53d0b0bcd8a292144ce5c1dc'), 1), (ObjectId('53d0b0bcd8a292144ce5c1de'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e0'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e1'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e3'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e4'), 1), (ObjectId('53d0b0bcd8a292144ce5c1e7'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ee'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ef'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f1'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f2'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f5'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f6'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f8'), 1), (ObjectId('53d0b0bcd8a292144ce5c1f9'), 1), (ObjectId('53d0b0bcd8a292144ce5c1fa'), 1), (ObjectId('53d0b0bcd8a292144ce5c1fd'), 1), (ObjectId('53d0b0bcd8a292144ce5c19d'), 1), (ObjectId('53d0b0bcd8a292144ce5c19e'), 1), (ObjectId('53d0b0bcd8a292144ce5c1a0'), 1), (ObjectId('53d0b0bcd8a292144ce5c1a3'), 1), (ObjectId('53d0b0bcd8a292144ce5c1a5'), 1), (ObjectId('53d0b0bcd8a292144ce5c1a9'), 1), (ObjectId('53d0b0bcd8a292144ce5c1aa'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ac'), 1), (ObjectId('53d0b0bcd8a292144ce5c1af'), 1), (ObjectId('53d0b0bcd8a292144ce5c1b2'), 1), (ObjectId('53d0b0bcd8a292144ce5c1b4'), 1), (ObjectId('53d0b0bcd8a292144ce5c1b5'), 1), (ObjectId('53d0b0bcd8a292144ce5c1b9'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ba'), 1), (ObjectId('53d0b0bcd8a292144ce5c1bb'), 1), (ObjectId('53d0b0bcd8a292144ce5c1bc'), 1), (ObjectId('53d0b0bcd8a292144ce5c1c2'), 1), (ObjectId('53d0b0bcd8a292144ce5c1c5'), 1), (ObjectId('53d0b0bcd8a292144ce5c1ca'), 1), (ObjectId('53d0b0bcd8a292144ce5c1d4'), 1), (ObjectId('53d0b0bcd8a292144ce5c1d5'), 1), (ObjectId('53d0b0bcd8a292144ce5c1d6'), 1), (ObjectId('53d0b0bcd8a292144ce5c1da'), 1), (ObjectId('53d0b0bcd8a292144ce5c1db'), 1)]
self.assertEqual(distribute_list_and_count_activity_index(smthng),{2.25: [ObjectId('53d0b0bbd8a292144ce5c191'), ObjectId('53d0b0bbd8a292144ce5c15a'), ObjectId('53d0b0bbd8a292144ce5c149'), ObjectId('53d0b0bbd8a292144ce5c147'), ObjectId('53d0b0bbd8a292144ce5c179'), ObjectId('53d0b0bbd8a292144ce5c163')], 2.5: [ObjectId('53d0b0bbd8a292144ce5c13d')], 3.0: [ObjectId('53d0b0bbd8a292144ce5c18a'), ObjectId('53d0b0bbd8a292144ce5c15d')], 2.9: [ObjectId('53d0b0bbd8a292144ce5c157'), ObjectId('53d0b0bbd8a292144ce5c17e')], 3.25: [ObjectId('53d0b0bbd8a292144ce5c151'), ObjectId('53d0b0bbd8a292144ce5c143'), ObjectId('53d0b0bbd8a292144ce5c175')], 2.65: [ObjectId('53d0b0bbd8a292144ce5c18b'), ObjectId('53d0b0bbd8a292144ce5c187'), ObjectId('53d0b0bbd8a292144ce5c148'), ObjectId('53d0b0bbd8a292144ce5c176')], 2.05: [ObjectId('53d0b0bbd8a292144ce5c198')], 3.3: [ObjectId('53d0b0bbd8a292144ce5c18f'), ObjectId('53d0b0bbd8a292144ce5c188')], 1.9: [ObjectId('53d0b0bbd8a292144ce5c17d')], 2.55: [ObjectId('53d0b0bbd8a292144ce5c181'), ObjectId('53d0b0bbd8a292144ce5c156'), ObjectId('53d0b0bbd8a292144ce5c14f'), ObjectId('53d0b0bbd8a292144ce5c145'), ObjectId('53d0b0bbd8a292144ce5c165'), ObjectId('53d0b0bbd8a292144ce5c166')], 3.4: [ObjectId('53d0b0bbd8a292144ce5c18c'), ObjectId('53d0b0bbd8a292144ce5c161'), ObjectId('53d0b0bbd8a292144ce5c162')], 2.45: [ObjectId('53d0b0bbd8a292144ce5c196'), ObjectId('53d0b0bbd8a292144ce5c193'), ObjectId('53d0b0bbd8a292144ce5c16d')], 3.55: [ObjectId('53d0b0bbd8a292144ce5c199'), ObjectId('53d0b0bbd8a292144ce5c150')], 2.1: [ObjectId('53d0b0bbd8a292144ce5c190'), ObjectId('53d0b0bbd8a292144ce5c17a'), ObjectId('53d0b0bbd8a292144ce5c169')], 1.3499999999999999: [ObjectId('53d0b0bbd8a292144ce5c195')], 3.1999999999999997: [ObjectId('53d0b0bbd8a292144ce5c192'), ObjectId('53d0b0bbd8a292144ce5c18e')], 3.15: [ObjectId('53d0b0bbd8a292144ce5c194'), ObjectId('53d0b0bbd8a292144ce5c141'), ObjectId('53d0b0bbd8a292144ce5c168')], 2.4: [ObjectId('53d0b0bbd8a292144ce5c185'), ObjectId('53d0b0bbd8a292144ce5c13b'), ObjectId('53d0b0bbd8a292144ce5c15c'), ObjectId('53d0b0bbd8a292144ce5c154'), ObjectId('53d0b0bbd8a292144ce5c155'), ObjectId('53d0b0bbd8a292144ce5c14d'), ObjectId('53d0b0bbd8a292144ce5c144'), ObjectId('53d0b0bbd8a292144ce5c171'), ObjectId('53d0b0bbd8a292144ce5c16a')], 2.2: [ObjectId('53d0b0bbd8a292144ce5c13f')], 2.9499999999999997: [ObjectId('53d0b0bbd8a292144ce5c197'), ObjectId('53d0b0bbd8a292144ce5c189'), ObjectId('53d0b0bbd8a292144ce5c15f'), ObjectId('53d0b0bbd8a292144ce5c158'), ObjectId('53d0b0bbd8a292144ce5c14a'), ObjectId('53d0b0bbd8a292144ce5c170'), ObjectId('53d0b0bbd8a292144ce5c164'), ObjectId('53d0b0bbd8a292144ce5c160')], 2.3: [ObjectId('53d0b0bbd8a292144ce5c183'), ObjectId('53d0b0bbd8a292144ce5c17f'), ObjectId('53d0b0bbd8a292144ce5c16c')], 3.5: [ObjectId('53d0b0bbd8a292144ce5c186')], 2.75: [ObjectId('53d0b0bbd8a292144ce5c17b'), ObjectId('53d0b0bbd8a292144ce5c177')], 1.3: [ObjectId('53d0b0bbd8a292144ce5c138')], 2.6999999999999997: [ObjectId('53d0b0bbd8a292144ce5c18d'), ObjectId('53d0b0bbd8a292144ce5c184'), ObjectId('53d0b0bbd8a292144ce5c182'), ObjectId('53d0b0bbd8a292144ce5c15b')], 3.9499999999999997: [ObjectId('53d0b0bbd8a292144ce5c17c')], 3.1: [ObjectId('53d0b0bbd8a292144ce5c19a'), ObjectId('53d0b0bbd8a292144ce5c13e'), ObjectId('53d0b0bbd8a292144ce5c13a'), ObjectId('53d0b0bbd8a292144ce5c15e'), ObjectId('53d0b0bbd8a292144ce5c174'), ObjectId('53d0b0bbd8a292144ce5c172'), ObjectId('53d0b0bbd8a292144ce5c16b')], 1.95: [ObjectId('53d0b0bbd8a292144ce5c159')], 2.6: [ObjectId('53d0b0bbd8a292144ce5c16e'), ObjectId('53d0b0bbd8a292144ce5c167')], 2.35: [ObjectId('53d0b0bbd8a292144ce5c13c'), ObjectId('53d0b0bbd8a292144ce5c139'), ObjectId('53d0b0bbd8a292144ce5c14c')], 2.85: [ObjectId('53d0b0bbd8a292144ce5c153'), ObjectId('53d0b0bbd8a292144ce5c14b')], 2.8: [ObjectId('53d0b0bbd8a292144ce5c180'), ObjectId('53d0b0bbd8a292144ce5c152'), ObjectId('53d0b0bbd8a292144ce5c14e'), ObjectId('53d0b0bbd8a292144ce5c146'), ObjectId('53d0b0bbd8a292144ce5c140'), ObjectId('53d0b0bbd8a292144ce5c142'), ObjectId('53d0b0bbd8a292144ce5c178'), ObjectId('53d0b0bbd8a292144ce5c173'), ObjectId('53d0b0bbd8a292144ce5c16f')]})
self.assertEqual(distribute_list_and_count_activity_index([]),{})
def test_select_similar(self):
self.assertEqual(select_no_of_users(ObjectId(),100),'Error: No Such Problem')
self.assertEqual(select_no_of_users(ObjectId(),0),'Error: No Such Problem')
self.assertEqual(select_no_of_users(ObjectId('53d0b0bbd8a292144ce5c198'),0),'Error: No Such Problem')
self.assertEqual(select_no_of_users(ObjectId('53d0b0bcd8a292144ce5c1c1'),0),[])
self.assertEqual(select_no_of_users(ObjectId('53d0b0bcd8a292144ce5c1c6'),1),[ObjectId('53d0b0bbd8a292144ce5c161')])
self.assertEqual(select_no_of_users(ObjectId('53d0b0bcd8a292144ce5c1c6'),2),[ObjectId('53d0b0bbd8a292144ce5c161'), ObjectId('53d0b0bbd8a292144ce5c186')])
self.assertEqual(select_no_of_users(ObjectId('53d0b0bcd8a292144ce5c1c6'),100),[ObjectId('53d0b0bbd8a292144ce5c161'), ObjectId('53d0b0bbd8a292144ce5c186'), ObjectId('53d0b0bbd8a292144ce5c192'), ObjectId('53d0b0bbd8a292144ce5c19a'), ObjectId('53d0b0bbd8a292144ce5c199'), ObjectId('53d0b0bbd8a292144ce5c18f'), ObjectId('53d0b0bbd8a292144ce5c18c'), ObjectId('53d0b0bbd8a292144ce5c178'), ObjectId('53d0b0bbd8a292144ce5c163'), ObjectId('53d0b0bbd8a292144ce5c13e'), ObjectId('53d0b0bbd8a292144ce5c152'), ObjectId('53d0b0bbd8a292144ce5c153'), ObjectId('53d0b0bbd8a292144ce5c189'), ObjectId('53d0b0bbd8a292144ce5c16e'), ObjectId('53d0b0bbd8a292144ce5c14c'), ObjectId('53d0b0bbd8a292144ce5c164'), ObjectId('53d0b0bbd8a292144ce5c167'), ObjectId('53d0b0bbd8a292144ce5c140'), ObjectId('53d0b0bbd8a292144ce5c17e'), ObjectId('53d0b0bbd8a292144ce5c17b'), ObjectId('53d0b0bbd8a292144ce5c168'), ObjectId('53d0b0bbd8a292144ce5c18b'), ObjectId('53d0b0bbd8a292144ce5c13a'), ObjectId('53d0b0bbd8a292144ce5c15f'), ObjectId('53d0b0bbd8a292144ce5c15b'), ObjectId('53d0b0bbd8a292144ce5c14d'), ObjectId('53d0b0bbd8a292144ce5c14b'), ObjectId('53d0b0bbd8a292144ce5c141'), ObjectId('53d0b0bbd8a292144ce5c16b'), ObjectId('53d0b0bbd8a292144ce5c151'), ObjectId('53d0b0bbd8a292144ce5c148'), ObjectId('53d0b0bbd8a292144ce5c143'), ObjectId('53d0b0bbd8a292144ce5c177'), ObjectId('53d0b0bbd8a292144ce5c171'), ObjectId('53d0b0bbd8a292144ce5c16c'), ObjectId('53d0b0bbd8a292144ce5c197'), ObjectId('53d0b0bbd8a292144ce5c198'), ObjectId('53d0b0bbd8a292144ce5c18d'), ObjectId('53d0b0bbd8a292144ce5c188'), ObjectId('53d0b0bbd8a292144ce5c17c'), ObjectId('53d0b0bbd8a292144ce5c16a'), ObjectId('53d0b0bbd8a292144ce5c162'), ObjectId('53d0b0bbd8a292144ce5c187'), ObjectId('53d0b0bbd8a292144ce5c182'), ObjectId('53d0b0bbd8a292144ce5c15e'), ObjectId('53d0b0bbd8a292144ce5c158'), ObjectId('53d0b0bbd8a292144ce5c156'), ObjectId('53d0b0bbd8a292144ce5c14e'), ObjectId('53d0b0bbd8a292144ce5c142'), ObjectId('53d0b0bbd8a292144ce5c16d'), ObjectId('53d0b0bbd8a292144ce5c16f'), ObjectId('53d0b0bbd8a292144ce5c179')])
if __name__ == '__main__':
unittest.main()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment