def test_clean_data_dict(self):
data = {'haha': [u'0,181', u'2', ], 'hoho': [u'0,181', u'2', ]}
self.assertEqual({'haha': [u'0', u'181', u'2', ], 'hoho': [u'0', u'181', u'2', ]}, clean_data_dict(data))
string_data = {'haha': u'2'}
self.assertEqual(string_data, clean_data_dict(string_data))
int_data = {'haha': 2}
self.assertEqual(int_data, clean_data_dict(int_data))
def test_clean_data_dict(self):
data = {"haha": [u"0,181", u"2"], "hoho": [u"0,181", u"2"]}
self.assertEqual({"haha": [u"0", u"181", u"2"], "hoho": [u"0", u"181", u"2"]}, clean_data_dict(data))
string_data = {"haha": u"2"}
self.assertEqual(string_data, clean_data_dict(string_data))
int_data = {"haha": 2}
self.assertEqual(int_data, clean_data_dict(int_data))
def test_clean_data_dict(self):
data = {'MultiChoice': [u'0,181', u'2', ], 'Text': [u'0,182', u'2', ], 'MultipleResponse': [u'0,181', u'2', u'3', u'4']}
self.assertEqual({'MultiChoice': [u'0', u'181', u'2', ], 'Text': [u'0', u'182', u'2', ],
'MultipleResponse': [u'0', u'181', [u'2', u'3', u'4']]}, clean_data_dict(data))
string_data = {'MultiChoice': u'2'}
self.assertEqual(string_data, clean_data_dict(string_data))
int_data = {'Number': 2}
self.assertEqual(int_data, clean_data_dict(int_data))
def test_get_number_of_extra_rows_supplied_for_an_answer_type(self):
data = {u'MultiChoice-MAX_NUM_FORMS': u'3', u'MultiChoice-TOTAL_FORMS': u'3',
u'MultiChoice-INITIAL_FORMS': u'3', u'MultiChoice-0-response': ['0,1', 1],
u'MultiChoice-1-response': ['1,1', 2, ], u'MultiChoice-2-response': ['2,1', 3, ],
u'MultipleResponse-MAX_NUM_FORMS': u'3', u'MultipleResponse-TOTAL_FORMS': u'3',
u'MultipleResponse-INITIAL_FORMS': u'3', u'MultipleResponse-0-response': ['0,1', '1', '2'],
u'MultipleResponse-1-response': ['1,1', '6', '7'], u'MultipleResponse-2-response': ['2,1', '3', '4'],
u'Number-MAX_NUM_FORMS': u'3', u'Number-TOTAL_FORMS': u'3',
u'Number-INITIAL_FORMS': u'3', u'Number-0-response': ['0,1', '22', ],
u'Number-1-response': ['1,1', '44', ], u'Number-2-response': ['2,1', '33', ],
u'Text-MAX_NUM_FORMS': u'3', u'Text-TOTAL_FORMS': u'3',
u'Text-INITIAL_FORMS': u'3', u'Text-0-response': ['0,1', 'Haha', ],
u'Text-1-response': ['1,1', 'Hehe', ], u'Text-2-response': ['2,1', 'hehehe', ],
u'Date-MAX_NUM_FORMS': u'3', u'Date-TOTAL_FORMS': u'3',
u'Date-INITIAL_FORMS': u'3', u'Date-0-response': ['0,1', '2014-2-2', ],
u'Date-1-response': ['1,1', '2014-2-2', ], u'Date-2-response': ['2,1', '2014-2-2', ]
}
data = clean_data_dict(dict(data))
self.assertEqual(['0', '1', '2'], extra_rows(data, 'Number', 1))
self.assertEqual(['0', '1', '2'], extra_rows(data, 'MultiChoice', 1))
self.assertEqual(['0', '1', '2'], extra_rows(data, 'Date', 1))
self.assertEqual(['0', '1', '2'], extra_rows(data, 'Text', 1))
self.assertEqual(['0', '1', '2'], extra_rows(data, 'MultipleResponse', 1))
def test_get_number_of_extra_rows_supplied_for_an_answer_type_when_there_are_more_than_one_grid_of_the_same_type(
self):
data = {u'MultiChoice-MAX_NUM_FORMS': u'6', u'MultiChoice-TOTAL_FORMS': u'6',
u'MultiChoice-INITIAL_FORMS': u'6', u'MultiChoice-0-response': ['0,1', 1, ],
u'MultiChoice-1-response': ['1,1', 2, ], u'MultiChoice-2-response': ['2,1', 3, ],
u'MultiChoice-3-response': ['0,2', 5, ],
u'MultiChoice-4-response': ['1,2', 6, ], u'MultiChoice-5-response': [2],
u'MultipleResponse-MAX_NUM_FORMS': u'6', u'MultipleResponse-TOTAL_FORMS': u'6',
u'MultipleResponse-INITIAL_FORMS': u'6', u'MultipleResponse-0-response': ['0,1', '1', '2'],
u'MultipleResponse-1-response': ['1,1', '6', '7'], u'MultipleResponse-2-response': ['1,2', '6', '7'],
u'MultipleResponse-3-response': ['0,2', '3', '4'], u'MultipleResponse-4-response': ['1,2', '3', '4'],
u'MultipleResponse-5-response': ['32', '41'],
u'Number-MAX_NUM_FORMS': u'3', u'Number-TOTAL_FORMS': u'3',
u'Number-INITIAL_FORMS': u'3', u'Number-0-response': ['0,1', '22', ],
u'Number-1-response': ['1,1', '44', ], u'Number-2-response': ['2,1', '33', ],
u'Text-MAX_NUM_FORMS': u'6', u'Text-TOTAL_FORMS': u'6',
u'Text-INITIAL_FORMS': u'6', u'Text-0-response': ['0,1', 'Haha', ],
u'Text-1-response': ['1,1', 'Hehe', ], u'Text-2-response': ['2,1', 'hehehe', ],
u'Text-3-response': ['0,2', 'Hehe1', ], u'Text-4-response': ['1,2', 'Hehe', ],
u'Text-5-response': ['hehe2'],
u'Date-MAX_NUM_FORMS': u'3', u'Date-TOTAL_FORMS': u'3',
u'Date-INITIAL_FORMS': u'3', u'Date-0-response': ['0,1', '2014-2-2', ],
u'Date-1-response': ['1,1', '2014-2-2', ], u'Date-2-response': ['2,1', '2014-2-2', ],
}
data = clean_data_dict(dict(data))
self.assertEqual(['0', '1', '2'], extra_rows(data, 'Number', 1))
self.assertEqual(['0', '1', '2'], extra_rows(data, 'MultiChoice', 1))
self.assertEqual(['0', '1'], extra_rows(data, 'MultiChoice', 2))
self.assertEqual(['0', '1', '2'], extra_rows(data, 'Date', 1))
self.assertEqual(['0', '1', '2'], extra_rows(data, 'Text', 1))
self.assertEqual(['0', '1'], extra_rows(data, 'Text', 2))
self.assertEqual(['0', '1'], extra_rows(data, 'MultipleResponse', 1))
self.assertEqual(['0', '1'], extra_rows(data, 'MultipleResponse', 2))
def test_gets_primary_answer_per_row_given_rows(self):
data = {
u'MultiChoice-MAX_NUM_FORMS': u'4', u'MultiChoice-TOTAL_FORMS': u'4', u'MultiChoice-INITIAL_FORMS': u'4',
u'MultiChoice-0-response': ['0,1', 'option1'],
u'MultiChoice-1-response': ['1,1', 'option2'],
u'MultiChoice-2-response': ['2,1', 'option3'],
u'MultiChoice-3-response': [4],
u'Number-MAX_NUM_FORMS': u'3', u'Number-TOTAL_FORMS': u'3', u'Number-INITIAL_FORMS': u'3',
u'Number-0-response': ['0,1', '22'],
u'Number-1-response': ['1,1', '44'],
u'Number-2-response': ['2,1', '33'],
u'Text-MAX_NUM_FORMS': u'4', u'Text-TOTAL_FORMS': u'4', u'Text-INITIAL_FORMS': u'4',
u'Text-0-response': ['0,1', 'row-0-column-0'],
u'Text-1-response': ['0,1', 'row-0-column-1'],
u'Text-2-response': ['1,1', 'row-1-column-0'],
u'Text-3-response': ['1,1', 'row1-column-1'],
u'Text-4-response': ['2,1', 'row-2-column-0'],
u'Text-5-response': ['2,1', 'row2-column-1'],
u'Date-MAX_NUM_FORMS': u'3', u'Date-TOTAL_FORMS': u'3', u'Date-INITIAL_FORMS': u'3',
u'Date-0-response': ['0,1', '2014-2-21'],
u'Date-1-response': ['1,1', '2014-2-22'],
u'Date-2-response': ['2,1', '2014-2-23'],
}
data = clean_data_dict(dict(data))
rows = extra_rows(data, "Number", group_id=1)
self.assertEqual(['row-0-column-0', 'row-1-column-0', 'row-2-column-0'],
primary_answers(data, rows, 'Text', group_id=1))
def test_get_extra_rows_when_type_is_repeated_per_row(self):
data = {
u"Number-1-response": [u"0,181", u"2"],
u"MultiChoice-MAX_NUM_FORMS": [u"6"],
u"redirect_url": [u""],
u"save_draft": [u""],
u"Number-6-response": [u"2,181", u"7"],
u"Number-INITIAL_FORMS": [u"3"],
u"Number-0-response": [u"0,181", u"1"],
u"MultiChoice-0-response": [u"0,181", u"96"],
u"MultiChoice-1-response": [u"0,181", u"115"],
u"Number-5-response": [u"1,181", u"6"],
u"Number-MAX_NUM_FORMS": [u"9"],
u"MultiChoice-4-response": [u"2,181", u"99"],
u"MultiChoice-TOTAL_FORMS": [u"6"],
u"Number-3-response": [u"1,181", u"4"],
u"Number-TOTAL_FORMS": [u"9"],
u"MultiChoice-3-response": [u"1,181", u"116"],
u"Number-2-response": [u"0,181", u"3"],
u"MultiChoice-5-response": [u"2,181", u"117"],
u"Number-7-response": [u"2,181", u"8"],
u"Number-8-response": [u"2,181", u"9"],
u"csrfmiddlewaretoken": [u"tx2R4QCrpl8IuHfaagFauUtC6XDs3u9C"],
u"MultiChoice-2-response": [u"1,181", u"98"],
u"MultiChoice-INITIAL_FORMS": [u"2"],
u"Number-4-response": [u"1,181", u"5"],
}
data = clean_data_dict(dict(data))
self.assertEqual(["0", "1", "2"], extra_rows(data, "Number", group_id=181))
self.assertEqual(["0", "1", "2"], extra_rows(data, "MultiChoice", group_id=181))
def test_get_number_of_extra_rows_supplied_for_an_answer_type(self):
data = {
u"MultiChoice-MAX_NUM_FORMS": u"3",
u"MultiChoice-TOTAL_FORMS": u"3",
u"MultiChoice-INITIAL_FORMS": u"3",
u"MultiChoice-0-response": ["0,1", 1],
u"MultiChoice-1-response": ["1,1", 2],
u"MultiChoice-2-response": ["2,1", 3],
u"Number-MAX_NUM_FORMS": u"3",
u"Number-TOTAL_FORMS": u"3",
u"Number-INITIAL_FORMS": u"3",
u"Number-0-response": ["0,1", "22"],
u"Number-1-response": ["1,1", "44"],
u"Number-2-response": ["2,1", "33"],
u"Text-MAX_NUM_FORMS": u"3",
u"Text-TOTAL_FORMS": u"3",
u"Text-INITIAL_FORMS": u"3",
u"Text-0-response": ["0,1", "Haha"],
u"Text-1-response": ["1,1", "Hehe"],
u"Text-2-response": ["2,1", "hehehe"],
u"Date-MAX_NUM_FORMS": u"3",
u"Date-TOTAL_FORMS": u"3",
u"Date-INITIAL_FORMS": u"3",
u"Date-0-response": ["0,1", "2014-2-2"],
u"Date-1-response": ["1,1", "2014-2-2"],
u"Date-2-response": ["2,1", "2014-2-2"],
}
data = clean_data_dict(dict(data))
self.assertEqual(["0", "1", "2"], extra_rows(data, "Number", 1))
self.assertEqual(["0", "1", "2"], extra_rows(data, "MultiChoice", 1))
self.assertEqual(["0", "1", "2"], extra_rows(data, "Date", 1))
self.assertEqual(["0", "1", "2"], extra_rows(data, "Text", 1))
def test_gets_primary_answer_in_a_row(self):
data = {'Number-0-response': [u'0,181', u'4', ], 'Number-1-response': [u'0,181', u'2'],
'Number-2-response': [u'0,181', u'3'], 'Text-0-response': [u'0,181', u'Haha']}
data = clean_data_dict(dict(data))
self.assertEqual('4',
_get_primary_answer_in('0', ['Number-0-response', 'Number-1-response', 'Number-2-response'],
data))
self.assertEqual(None, _get_primary_answer_in('4', ['Text-0-response'], data))
def __init__(self, section, initial={}, data=None, highlight=False, edit_after_submit=False):
self.initial = initial
self.data = data
self.cleaned_data = clean_data_dict(dict(copy.deepcopy(data))) if data else None
self.section = section
self.mapped_question_orders = section.mapped_question_orders()
self.country = self.initial.get('country')
self.version = self.initial.get('version')
self.questionnaire = self.initial.get('questionnaire')
self.extra_rows = {}
self.formsets = self._formsets()
self.ANSWER_FORM_COUNTER = self._initialize_form_counter()
self._highlight_required_answers(highlight)
self.form_answer_map = {}
self.editing_after_submit = edit_after_submit
def test_get_extra_rows_when_type_is_repeated_per_row(self):
data = {u'Number-1-response': [u'0,181', u'2', ], u'MultiChoice-MAX_NUM_FORMS': [u'6'], u'redirect_url': [u''],
u'save_draft': [u''], u'Number-6-response': [u'2,181', u'7', ], u'Number-INITIAL_FORMS': [u'3'],
u'Number-0-response': [u'0,181', u'1', ], u'MultiChoice-0-response': [u'0,181', u'96', ],
u'MultiChoice-1-response': [u'0,181', u'115', ], u'Number-5-response': [u'1,181', u'6', ],
u'Number-MAX_NUM_FORMS': [u'9'], u'MultiChoice-4-response': [u'2,181', u'99', ],
u'MultiChoice-TOTAL_FORMS': [u'6'], u'Number-3-response': [u'1,181', u'4', ],
u'Number-TOTAL_FORMS': [u'9'], u'MultiChoice-3-response': [u'1,181', u'116', ],
u'Number-2-response': [u'0,181', u'3', ], u'MultiChoice-5-response': [u'2,181', u'117', ],
u'Number-7-response': [u'2,181', u'8', ], u'Number-8-response': [u'2,181', u'9', ],
u'csrfmiddlewaretoken': [u'tx2R4QCrpl8IuHfaagFauUtC6XDs3u9C'],
u'MultiChoice-2-response': [u'1,181', u'98', ], u'MultiChoice-INITIAL_FORMS': [u'2'],
u'Number-4-response': [u'1,181', u'5', ]}
data = clean_data_dict(dict(data))
self.assertEqual(['0', '1', '2'], extra_rows(data, "Number", group_id=181))
self.assertEqual(['0', '1', '2'], extra_rows(data, "MultiChoice", group_id=181))
def test_get_number_of_extra_rows_supplied_for_an_answer_type_when_there_are_more_than_one_grid_of_the_same_type(
self
):
data = {
u"MultiChoice-MAX_NUM_FORMS": u"6",
u"MultiChoice-TOTAL_FORMS": u"6",
u"MultiChoice-INITIAL_FORMS": u"6",
u"MultiChoice-0-response": ["0,1", 1],
u"MultiChoice-1-response": ["1,1", 2],
u"MultiChoice-2-response": ["2,1", 3],
u"MultiChoice-3-response": ["0,2", 5],
u"MultiChoice-4-response": ["1,2", 6],
u"MultiChoice-5-response": [2],
u"Number-MAX_NUM_FORMS": u"3",
u"Number-TOTAL_FORMS": u"3",
u"Number-INITIAL_FORMS": u"3",
u"Number-0-response": ["0,1", "22"],
u"Number-1-response": ["1,1", "44"],
u"Number-2-response": ["2,1", "33"],
u"Text-MAX_NUM_FORMS": u"6",
u"Text-TOTAL_FORMS": u"6",
u"Text-INITIAL_FORMS": u"6",
u"Text-0-response": ["0,1", "Haha"],
u"Text-1-response": ["1,1", "Hehe"],
u"Text-2-response": ["2,1", "hehehe"],
u"Text-3-response": ["0,2", "Hehe1"],
u"Text-4-response": ["1,2", "Hehe"],
u"Text-5-response": ["hehe2"],
u"Date-MAX_NUM_FORMS": u"3",
u"Date-TOTAL_FORMS": u"3",
u"Date-INITIAL_FORMS": u"3",
u"Date-0-response": ["0,1", "2014-2-2"],
u"Date-1-response": ["1,1", "2014-2-2"],
u"Date-2-response": ["2,1", "2014-2-2"],
}
data = clean_data_dict(dict(data))
self.assertEqual(["0", "1", "2"], extra_rows(data, "Number", 1))
self.assertEqual(["0", "1", "2"], extra_rows(data, "MultiChoice", 1))
self.assertEqual(["0", "1"], extra_rows(data, "MultiChoice", 2))
self.assertEqual(["0", "1", "2"], extra_rows(data, "Date", 1))
self.assertEqual(["0", "1", "2"], extra_rows(data, "Text", 1))
self.assertEqual(["0", "1"], extra_rows(data, "Text", 2))
def test_gets_primary_answer_per_row_given_rows(self):
data = {
u"MultiChoice-MAX_NUM_FORMS": u"4",
u"MultiChoice-TOTAL_FORMS": u"4",
u"MultiChoice-INITIAL_FORMS": u"4",
u"MultiChoice-0-response": ["0,1", "option1"],
u"MultiChoice-1-response": ["1,1", "option2"],
u"MultiChoice-2-response": ["2,1", "option3"],
u"MultiChoice-3-response": [4],
u"Number-MAX_NUM_FORMS": u"3",
u"Number-TOTAL_FORMS": u"3",
u"Number-INITIAL_FORMS": u"3",
u"Number-0-response": ["0,1", "22"],
u"Number-1-response": ["1,1", "44"],
u"Number-2-response": ["2,1", "33"],
u"Text-MAX_NUM_FORMS": u"4",
u"Text-TOTAL_FORMS": u"4",
u"Text-INITIAL_FORMS": u"4",
u"Text-0-response": ["0,1", "row-0-column-0"],
u"Text-1-response": ["0,1", "row-0-column-1"],
u"Text-2-response": ["1,1", "row-1-column-0"],
u"Text-3-response": ["1,1", "row1-column-1"],
u"Text-4-response": ["2,1", "row-2-column-0"],
u"Text-5-response": ["2,1", "row2-column-1"],
u"Date-MAX_NUM_FORMS": u"3",
u"Date-TOTAL_FORMS": u"3",
u"Date-INITIAL_FORMS": u"3",
u"Date-0-response": ["0,1", "2014-2-21"],
u"Date-1-response": ["1,1", "2014-2-22"],
u"Date-2-response": ["2,1", "2014-2-23"],
}
data = clean_data_dict(dict(data))
rows = extra_rows(data, "Number", group_id=1)
self.assertEqual(
["row-0-column-0", "row-1-column-0", "row-2-column-0"], primary_answers(data, rows, "Text", group_id=1)
)
