def test_grouped_continuous(self):
gcat = Question(self.df[['q2_1', 'q2_2']], 'This is a categorical question?',
statements=['Statement A', 'Statement B'])
self.assertEqual(gcat.type, 'grouped_continuous')
self.assertEqual(gcat.question, 'This is a categorical question?')
self.assertEqual(gcat.responses, None)
self.assertEqual(gcat.data.shape[0], 1000)
def test_categorical(self):
cat = Question(self.df.q2_1, 'This is a categorical question?', responses=['A', 'B', 'C', 'D', 'E'])
self.assertEqual(cat.type, 'categorical')
self.assertEqual(cat.question, 'This is a categorical question?')
self.assertEqual(cat.responses, ['A', 'B', 'C', 'D', 'E'])
self.assertEqual(list(cat.data.categories), ['A', 'B', 'C', 'D', 'E'])
self.assertEqual(cat.data.shape[0], 1000)
def test_grouped_categorical(self):
gcat = Question(self.df[['q2_1', 'q2_2']], 'This is a categorical question?',
responses=['A', 'B', 'C', 'D', 'E'], statements=['Statement A', 'Statement B'])
self.assertEqual(gcat.type, 'grouped_categorical')
self.assertEqual(gcat.question, 'This is a categorical question?')
self.assertEqual(gcat.responses, ['A', 'B', 'C', 'D', 'E'])
self.assertEqual(gcat.data.shape[0], 1000)
def test_continuous(self):
cont = Question(self.df.q3_1, 'This is a continuous question?')
self.assertEqual(cont.question, 'This is a continuous question?')
self.assertEqual(cont.type, 'continuous')
self.assertEqual(cont.responses, None)
self.assertEqual(cont.data.shape[0], 1000)
def test_date_fail(self):
with self.assertRaises(ValueError):
Question(pd.Series([1,2,3,4,5]), question='Dummy?',date=True)
def test_date(self):
dates = pd.date_range(pd.datetime.today(), periods=30, freq='D').to_list()
dates = pd.Series(dates).sample(1000, replace=True)
dat = Question(dates, 'This is a date question?', date=True)
self.assertEqual(dat.type, 'datetime')
self.assertEqual(dat.data.shape[0], 1000)
def test_multipunch_fail(self):
with self.assertRaises(ValueError):
Question(pd.Series([1,2,3,4,5]), question='Dummy?',multipunch=True)
def test_multipunch(self):
multi = Question(self.df[['q5_1', 'q5_2']], 'This is a multipunch question?', multipunch=True)
self.assertEqual(multi.type, 'multipunch')
self.assertTrue(multi.data.shape[1] > 0)
self.assertEqual(multi.data.shape[0], 1000)
def _validate(self, item):
if not isinstance(item, type(Question())):
raise AttributeError 'Not all survey elements are question objects'
else:
pass