r_u_t_fn = generate_repo_user_times(args.version, r_to_i, u_to_i,
is_test=args.test)
tr_fn, v_fn, rt_fn = generate_training_validating_rt(
args.version, r_to_i, u_to_i, r_u_t_fn,
args.split, is_test=args.test)
f_fn = generate_followerships(args.version, u_to_i, is_test=args.test)
present = time.time()
print "Wall clock time: %.3f s" % (present - past)
if args.test:
from nose.tools import ok_, eq_
from scipy.io import mmread
print "u_to_i",u_to_i
print "r_to_i",r_to_i
ok_( [1,2,3,5,6] == u_to_i.keys() )
ok_( range(1,12) == r_to_i.keys() )
eq_(mmread(r_u_t_fn).nnz, 26)
eq_(mmread(tr_fn).nnz, 20)
eq_(mmread(v_fn).nnz, 7)
rt = np.load(rt_fn)
ok_( np.all(set(rt) == set([160000,198000,300000,202000,210000])) )
foll = mmread(f_fn).tocsr()
eq_(foll.nnz, 6)
eq_(foll[u_to_i[1],u_to_i[2]], 122000)
ok_( np.all((foll.sum(0).A1+foll.sum(1).A1) >= 1) )
print "Tests Pass"
