def test1(self):
R = DataFrame([('SUBJECT', [1, 2]), ('TIMEOFDAY', [u'T1', u'T1']),
('COURSE', [u'C1', u'C2']), ('MODEL', [u'M1', u'M1']),
('ERROR', [10, 10])])
df = DataFrame()
df.read_tbl('data/error~subjectXtimeofdayXcourseXmodel_MISSING.csv')
df2 = df.where(['ERROR = 10'])
self.assertEqual(repr(df2), repr(R))
def test5(self):
R = DataFrame([
('SUBJECT', [1, 1, 1, 1, 1, 1, 2, 2, 2, 3, 3, 3, 3, 3, 3]),
('TIMEOFDAY', [
u'T1', u'T1', u'T1', u'T2', u'T2', u'T2', u'T2', u'T2', u'T2',
u'T1', u'T1', u'T1', u'T2', u'T2', u'T2'
]),
('COURSE', [
u'C1', u'C1', u'C1', u'C1', u'C1', u'C1', u'C1', u'C1', u'C1',
u'C1', u'C1', u'C1', u'C1', u'C1', u'C1'
]),
('MODEL', [
u'M1', u'M2', u'M3', u'M1', u'M2', u'M3', u'M1', u'M2', u'M3',
u'M1', u'M2', u'M3', u'M1', u'M2', u'M3'
]), ('ERROR', [10, 8, 6, 5, 4, 3, 4, 3, 3, 8, 7, 4, 4, 1, 2])
])
df = DataFrame()
df.read_tbl('data/error~subjectXtimeofdayXcourseXmodel_MISSING.csv')
df2 = df.where(['COURSE = "C1"', 'TIMEOFDAY in ("T1", "T2")'])
self.assertEqual(repr(df2), repr(R))
