def test_n_cas_con_flag(self): self.args.N = None self.args.N_cas = 1000.0 self.args.N_con = 234.0 dat = munge.process_n(self.dat, self.args, log) assert_series_equal(dat.N, self.N_const, check_names=False)
def test_nstudy(self): # should filter on NSTUDY if the --N flag is set, but N gets set to # 9999 self.dat['NSTUDY'] = self.N dat = munge.process_n(self.dat, self.args, log) assert_frame_equal(dat, self.dat_filtered9999)
def test_n_cas_con_col(self): self.dat['N_CAS'] = self.N self.dat['N_CON'] = [0.0, 0, 0] dat = munge.process_n(self.dat, self.args, log) assert_frame_equal(dat, self.dat_filtered)
def test_n_col(self): self.dat['N'] = self.N dat = munge.process_n(self.dat, self.args, log) print dat print self.dat_filtered assert_frame_equal(dat, self.dat_filtered)
def test_n_flag(self): self.args.N = 1234.0 self.args.N_cas = None self.args.N_con = None dat = munge.process_n(self.dat, self.args, log) self.assertTrue(set(dat.N) == set(self.N_const))
def test_n_flag(self): self.args.N = 1234.0 self.args.N_cas = None self.args.N_con = None dat = munge.process_n(self.dat, self.args, log) assert_series_equal(dat.N, self.N_const)