def test__liteqclass(self): u = 1 set_of_u_1 = zkl.load("{}/liteqclass_output_n_5_u_1.zkl".format(self._ABS_PATH)) g2 = bfutils.undersample(self._G, u) s = ur.liteqclass(g2, verbose=False, capsize=1000) self.assertEqual(s, set_of_u_1) u = 2 set_of_u_2 = zkl.load("{}/liteqclass_output_n_5_u_2.zkl".format(self._ABS_PATH)) g2 = bfutils.undersample(self._G, u) s = ur.liteqclass(g2, verbose=False, capsize=1000) self.assertEqual(s, set_of_u_2)
def ra_wrapper_preset(fold, glist=[]): scipy.random.seed() l = {} while True: try: g = glist[fold] gs = bfutils.call_undersamples(g) for u in range(1, min([len(gs), UMAX])): g2 = bfutils.undersample(g, u) print fold, ': ', traversal.density(g), ':', startTime = int(round(time.time() * 1000)) s = ur.liteqclass(g2, verbose=False, capsize=CAPSIZE) endTime = int(round(time.time() * 1000)) print len(s), u l[u] = {'eq': s, 'ms': endTime - startTime} except MemoryError: print 'memory error... retrying' continue break return {'gt': g, 'solutions': l}