Example #1
0
 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)
Example #2
0
 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)
Example #3
0
 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)
Example #4
0
 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)
Example #5
0
 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)
Example #6
0
 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)
Example #7
0
 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)
Example #8
0
 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)
Example #9
0
 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))
Example #10
0
 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)