def test_inv_wo_test_id(self): inv_test = INV(**self.dummy_test_data, expect=Expect.eq(), name="inv test without test id") assert inv_test.test_id is None
def object_test(): global food_ret, sport_ret, drug_ret, nondrug_ret """ codeDictionary = {"D":0, "M":1, "S":2, "H":3, "F":4, "O":5, "E":6, "NA":7} """ editor = Editor() food_ret1 = editor.template('How often do you get {food}?', food=food, labels=0, save=True) #, nsamples=100) a food_ret2 = editor.template('I can\'t stop thinking about {food}!', food=food, labels=0, save=True) #, nsamples=100) food_ret = food_ret1 + food_ret2 mft_food = MFT(food_ret.data, labels=food_ret.labels, name='Object Rec: Food', capability='Objects', description='Food') sport_ret1 = editor.template('I have to participate in {sport}?', sport=sport, labels=6, save=True) #, nsamples=100) sport_ret2 = editor.template( 'It is good to move your body, like doing {sport}.', sport=sport, labels=6, save=True) #, nsamples=100) sport_ret = sport_ret1 + sport_ret2 mft_sport = MFT(sport_ret.data, labels=sport_ret.labels, name='Object Rec: Sport', capability='Objects', description='Sport') nondrug_ret1 = editor.template('How often do you take {nondrug}?', nondrug=nondrug, labels=5) #, save=True) #, nsamples=100) nondrug_ret2 = editor.template( 'Have you taken {nondrug} for the last five months?', nondrug=nondrug, labels=5) #, save=True) #, nsamples=100) nondrug_ret = nondrug_ret1 + nondrug_ret2 mft_nondrug = MFT(nondrug_ret.data, labels=nondrug_ret.labels, name='Object Rec: Non Drug', capability='Objects', description='Non Drug') drug_ret1 = editor.template('How often do you get {drug}?', drug=drug, labels=1, save=True) #, nsamples=100) drug_ret2 = editor.template( 'Have you taken {drug} for the last five months?', drug=drug, labels=1, save=True) #, nsamples=100) drug_ret = drug_ret1 + drug_ret2 mft_drug = MFT(drug_ret.data, labels=drug_ret.labels, name='Object Rec: Drug', capability='Objects', description='Drug') #print(nondrug_ret.data) nt = Perturb.perturb(nondrug_ret.data, swap_nondrug) inv_n = INV(**nt, name='swap nondrug name in both questions', capability='objects', description='') #print(len(nt.data)) #exit() """ import numpy as np def pp(inputs): p1 = np.array([0.5 for x in inputs]).reshape(-1, 1) p0 = 1- p1 return np.hstack((p0, p1)) from checklist.pred_wrapper import PredictorWrapper wrapped = PredictorWrapper.wrap_softmax(pp) inv_n.run(wrapped) """ dt = Perturb.perturb(drug_ret.data, swap_drug) inv_d = INV(**dt, name='swap drug name in both questions', capability='objects', description='') nondrug_monodec = Expect.monotonic(label=5, increasing=False, tolerance=0.1) drug_monodec = Expect.monotonic(label=1, increasing=False, tolerance=0.1) ndt = Perturb.perturb(nondrug_ret.data, swap_nd) dir_nd = DIR(**ndt, expect=nondrug_monodec) dnt = Perturb.perturb(drug_ret.data, swap_dn) dir_dn = DIR(**dnt, expect=drug_monodec) # diet #exercise # other # medical # other # medical, # o -> m, # m->o tests = [mft_food, mft_sport, mft_nondrug, mft_drug] #, inv_n , inv_d, dir_nd, dir_dn ] names = [ x.strip(",") for x in "mft_food, mft_sport, mft_nondrug, mft_drug".split() ] #, inv_n, inv_d, dir_nd, dir_dn".split() ] assert (len(tests) == len(names)) for test, name in zip(tests, names): test.to_raw_file('./tests/' + name + '.txt') return tests, names
def test_inv_w_test_id(self): inv_test = INV(**self.dummy_test_data, expect=Expect.eq(), name="inv test with test id", test_id=self.test_id) assert inv_test.test_id == self.test_id
def test_dir_wo_test_id(self): dir_test = DIR(**self.dummy_test_data, expect=Expect.eq(), name="dir test without test id") assert dir_test.test_id is None
def test_dir_w_test_id(self): dir_test = DIR(**self.dummy_test_data, expect=Expect.eq(), name="dir test with test id", test_id=self.test_id) assert dir_test.test_id == self.test_id
def test_mft_wo_test_id(self): mft_test = MFT(**self.dummy_test_data, expect=Expect.eq(), name="mft test without test id") assert mft_test.test_id is None
def test_mft_w_test_id(self): mft_test = MFT(**self.dummy_test_data, expect=Expect.eq(), name="mft test with test id", test_id=self.test_id) assert mft_test.test_id == self.test_id