def test__FilterManufacturedEvents_returns_correct_events_and_index(self): results = [r for r in self.mock_results if r['survey_type'] != 'setup.js'] results = processresults._DiscardResultsBeforeDate( results, datetime.datetime(2014, 12, 01, 0, 0, 0, 0)) results, canonical_index = processresults._FilterManufacturedEvents(results) self.assertEqual(len(results), 2) self.assertEqual(canonical_index, 1)
def test__DiscardResultsBeforeDate_filters_out_two_november_dates(self): results = [ r for r in self.mock_results if (r['survey_type'] != 'setup.js' and r['responses'][0]['question'] != 'MANUFACTURED')] results = processresults._DiscardResultsBeforeDate( results, datetime.datetime(2014, 12, 01, 0, 0, 0, 0)) self.assertEqual(len(results), 3) self.assertEqual(results[0]['date_taken'], u'2015-01-05T11:48:39.760000') self.assertEqual(results[1]['date_taken'], u'2015-01-05T00:00:00.123456') self.assertEqual(results[2]['date_taken'], u'2015-01-05T01:02:03.123456')