def test_tet_isroot1(self): ''' test isroot true in tet ''' username = '******' test_objekt = tet.TET(root=username) self.assertTrue(test_objekt.isroot(username))
def test_tet_tostring1(self): ''' test tostring in simple tet ''' username = '******' test_objekt = tet.TET(root=username) self.assertEqual(test_objekt.tostring(), '[u:1234]')
def test_tet_isroot2(self): ''' test isroot false in tet ''' username = '******' test_objekt = tet.TET(root=username) self.assertFalse(test_objekt.isroot('lars'))
def test_tet_count_children1(self): ''' test count_children in tet without children ''' username = '******' test_objekt = tet.TET(root=username) self.assertEqual(test_objekt.count_children(), {})
def test_tet_getroot(self): ''' tests getroot in tet ''' username = '******' test_objekt = tet.TET(root=username) self.assertIs(test_objekt.getroot(), username)
def load_tets(loadpath, limit=None): ''' description: This function loads and rebuilds the tets saved by save_tets parameters: the loadpath is the path to the file where the tets are saved return: the return is a dictionary of all tets ''' tets = {} count = 1 with open(loadpath, 'r', encoding="utf-8") as file: for stringtet in tqdm(csv.reader(file)): tetchildren = [] for stringsubtet in stringtet[1:]: stringsubtet = stringsubtet.replace('[', '').replace(']', '').split(':') stringsubtet[0] = stringsubtet[0].split(',') genres = [] for genre in stringsubtet[0][1:]: genres.append(tet.TETChild(genre)) partlist = [tet.TETChild(stringsubtet[0][0], children=genres)]\ * int(stringsubtet[1]) tetchildren = tetchildren + partlist tets[stringtet[0]] = tet.TET(stringtet[0], children=tetchildren) if limit is not None: if count >= limit: break count += 1 return tets
def test_tet_getchildren(self): ''' test get children in tet ''' username = '******' children = [tet.TETChild('Lars')] test_objekt = tet.TET(root=username, children=children) self.assertEqual(test_objekt.getchildren(), children)
def test_tet_count_children2(self): ''' test count_children in tet with children ''' username = '******' children = [tet.TETChild('high', children=tet.TETChild('action'))] test_objekt = tet.TET(root=username, children=children) self.assertEqual(test_objekt.count_children(), {'[high,[[action]]]': 1})
def test_tet_tostring2(self): ''' test tostring in tet with children ''' username = '******' children = [tet.TETChild('high', children=tet.TETChild('action'))] test_objekt = tet.TET(root=username, children=children) self.assertEqual(test_objekt.tostring(), '[u:1234,[[high,[[action]]]:1]]')
def tet_find_tree(user, tets): ''' description: This function finds the TET that has user as the root if such a tree does not exist it is build parameters: user is a user id as a string and tets is the dictionary of tets return: is a TET with user as root ''' if user in tets: return tets[user] return tet.TET(root=user)
def test_tet_find_most_with_rating5(self): ''' test find_most_with_rating in tet ''' username = '******' children = [ tet.TETChild('low', children=tet.TETChild('action')), tet.TETChild('low', children=tet.TETChild('comedy')) ] test_objekt = tet.TET(root=username, children=children) self.assertEqual(test_objekt.find_most_with_rating('high'), [['[high,[[nohigh]]]', 0]])
def test_tet_find_tree2(self): ''' tests the find_tree where the tet does not exists ''' username = '******' children = [ tet.TETChild('high', children=tet.TETChild('action')), tet.TETChild('high', children=tet.TETChild('action')) ] test_tet = tet.TET(root=username, children=children) test_tets = {username: test_tet} self.assertNotEqual(build_tet.tet_find_tree('u:5678', test_tets), test_tet)
def test_tet_construction(self): ''' test the construction of a tet ''' test_objekt = tet.TET() self.assertIsInstance(test_objekt, tet.TET)