def test_recap_document_already_exists(self, mock): """We already have everything""" # Update self.rd so it looks like it is already all good. self.rd.is_available = True cf = ContentFile(self.file_content) self.rd.filepath_local.save(self.filename, cf) rd = process_recap_pdf(self.pq.pk) # Did we avoid creating new objects? self.assertEqual(rd, self.rd) self.assertEqual(rd.docket_entry, self.de) self.assertEqual(rd.docket_entry.docket, self.docket) # Did we update pq appropriately? self.pq.refresh_from_db() self.assertEqual(self.pq.status, self.pq.PROCESSING_SUCCESSFUL) self.assertEqual(self.pq.error_message, 'Successful upload! Nice work.') self.assertFalse(self.pq.filepath_local) self.assertEqual(self.pq.docket_id, self.docket.pk) self.assertEqual(self.pq.docket_entry_id, self.de.pk) self.assertEqual(self.pq.recap_document_id, self.rd.pk) # Did we correctly avoid running document extraction? mock.assert_not_called()
def test_close_is_called(self, mock, full_path): loader = TensorflowRecordWriter() loader.open(str( full_path)) # use string here since Tf_records to not use pathlib mock.assert_not_called() loader.close() mock.assert_called_once()
def test_get_train_example(self, mock, mock_image_data): mock_image_data.bit_depth = 8 mock_image_data.matrix = numpy.array([1, 2, 3]) factory = TensorflowTrainExamplesFromImageData() factory.set_image_data(mock_image_data) mock.assert_not_called() factory.get_train_example() mock.assert_called_once()
def test_opbeat_tween_factory_returns_a_curried_tween_function(self, mock): handler = {} registry = self.request.registry result = subscribers.opbeat_tween_factory(handler, registry) mock.assert_not_called() result(self.request) mock.assert_called_once_with(handler, registry, self.request)
def test_write_is_called(self, mock, full_path): loader = TensorflowRecordWriter() loader.open(str( full_path)) # use string here since Tf_records to not use pathlib mock.assert_not_called() raw_data = numpy.array([1, 1, 2, 3, 4]).astype(numpy.int16).tobytes() feature = tensorflow.train.Feature( bytes_list=tensorflow.train.BytesList(value=[raw_data])) result = {"feature": feature} example = tensorflow.train.Example(features=tensorflow.train.Features( feature=result)) loader.write(example) mock.assert_called_once()
def test_debug_does_not_create_rd(self, mock): """If debug is passed, do we avoid creating recap documents?""" docket = Docket.objects.create(source=0, court_id='scotus', pacer_case_id='asdf') DocketEntry.objects.create(docket=docket, entry_number=1) pq = ProcessingQueue.objects.create( court_id='scotus', uploader=self.user, pacer_case_id='asdf', pacer_doc_id='asdf', document_number='1', filepath_local=self.pdf, upload_type=ProcessingQueue.PDF, debug=True, ) _ = process_recap_pdf(pq.pk) self.assertEqual(RECAPDocument.objects.count(), 0) mock.assert_not_called()
def test_recap_document_already_exists(self, mock): """We already have everything""" # Update self.rd so it looks like it is already all good. self.rd.is_available = True cf = ContentFile(self.file_content) self.rd.filepath_local.save(self.filename, cf) rd = process_recap_pdf(self.pq.pk) # Did we avoid creating new objects? self.assertEqual(rd, self.rd) self.assertEqual(rd.docket_entry, self.de) self.assertEqual(rd.docket_entry.docket, self.docket) # Did we update pq appropriately? self.pq.refresh_from_db() self.assertEqual(self.pq.status, self.pq.PROCESSING_SUCCESSFUL) self.assertFalse(self.pq.error_message) self.assertFalse(self.pq.filepath_local) # Did we correctly avoid running document extraction? mock.assert_not_called()
def test_debug_does_not_create_recap_documents(self, mock): """If debug is passed, do we avoid creating recap documents?""" d = Docket.objects.create(source=0, court_id='scotus', pacer_case_id='asdf') de = DocketEntry.objects.create(docket=d, entry_number=1) rd = RECAPDocument.objects.create( docket_entry=de, document_number='1', pacer_doc_id='04505578698', document_type=RECAPDocument.PACER_DOCUMENT, ) pq = ProcessingQueue.objects.create( court_id='scotus', uploader=self.user, upload_type=ProcessingQueue.ATTACHMENT_PAGE, filepath_local=self.att, debug=True, ) process_recap_attachment(pq.pk) self.assertEqual(Docket.objects.count(), 1) self.assertEqual(DocketEntry.objects.count(), 1) self.assertEqual(RECAPDocument.objects.count(), 1) mock.assert_not_called()
def test_assign_label(self, mock, mock_label, mock_image_data): label_image_data = LabelImageData() label_image_data.set_data(mock_image_data) mock.assert_not_called() label_image_data.assign(mock_label) mock.assert_called_once()
def test_open(self, mock, full_path): loader = TensorflowRecordWriter() mock.assert_not_called() loader.open(full_path) mock.assert_called_once()
def test_close_is_called(self, mock, full_path, create_tmp_file): loader = PillowImageLoader() loader.open(full_path) mock.assert_not_called() loader.close() mock.assert_called_once()
def test_read_is_called(self, mock, full_path, create_tmp_file): loader = PillowImageLoader() mock.assert_not_called() loader.get_data() mock.assert_called_once()