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()
