def test_pre_process(self): after_octo = {'boat':0,'dress':1,'shirt':2,'pants':3,'suit':4,'food':5,'hat':6,'watch':7,'car':8,'plane':9,'frills':10,'seacreatures':11,'bananas':12,'glasses':13} #pass in 1 for it to be sorted user_pos,item_pos,item_user_dict,user_dict = rec.pre_process(self.true_parsed_customers,1) for k in after_octo: self.assertEqual(after_octo[k],item_pos[k]) for k in item_pos: self.assertEqual(after_octo[k],item_pos[k])
def test_create_sparse_mat(self): user_pos,item_pos,item_user_dict,user_dict= rec.pre_process(self.true_parsed_customers,1) ret = rec.create_sparse_user_item_mat(item_user_dict,len(item_pos)) self.assertTrue(np.array_equal(np.array(self.true_matrix),ret.toarray())) print ret.toarray()