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