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 _assert_send(self, mock, title, body):
mock.assert_called_once()
send_args, send_kwargs = mock.call_args
message = send_args[-1]({'token': 'XYZ_TOKEN'})
self.assertEqual(message.notification.title, title)
self.assertEqual(message.notification.body, body)
def testMissingTypeError(self, mock):
config = dict(Extensions=dict(disable_defaults=True, foo=dict()))
ext = _yaml_load_extensions(config)
self.assertEqual(ext, [])
mock.assert_called_once()
args, _ = mock.call_args
self.assertIn("The section '%s' must contain a 'type' parameter.", args[0])
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 testUpdateParameters(self):
""" sigproc.fit.Model update_parameter """
self.model.update_parameter(parameter='p1_0', value=1.5, min=0.0, max=2.5, expr='<=2.5')
self.assertEqual(self.model.parameters['p1_0'].value, 1.5)
self.assertEqual(self.model.parameters['p1_0'].min, 0.0)
self.assertEqual(self.model.parameters['p1_0'].max, 2.5)
self.assertEqual(self.model.parameters['p1_0'].expr, '<=2.5')
with patch.object(self.model, 'push_parameters') as mock:
self.model.update_parameter(parameter='p1_0', value=2.5)
mock.assert_called_once()
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 testUpdateParameters(self):
""" sigproc.fit.Model update_parameter """
self.model.update_parameter(parameter='p1_0',
value=1.5,
min=0.0,
max=2.5,
expr='<=2.5')
self.assertEqual(self.model.parameters['p1_0'].value, 1.5)
self.assertEqual(self.model.parameters['p1_0'].min, 0.0)
self.assertEqual(self.model.parameters['p1_0'].max, 2.5)
self.assertEqual(self.model.parameters['p1_0'].expr, '<=2.5')
with patch.object(self.model, 'push_parameters') as mock:
self.model.update_parameter(parameter='p1_0', value=2.5)
mock.assert_called_once()
def test_docket_and_docket_entry_already_exist(self, mock):
"""What happens if we have everything but the PDF?
This is the good case. We simply create a new item.
"""
self.rd.delete()
rd = process_recap_pdf(self.pq.pk)
self.assertTrue(rd.is_available)
self.assertTrue(rd.sha1)
self.assertTrue(rd.filepath_local)
mock.assert_called_once()
self.assertIn('gov.uscourts.scotus.asdf.1.0', rd.filepath_local.name)
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)
def testCallError(self, mock):
node = base.NodeBase('root', None)
node(0)
mock.assert_called_once()
args, _ = mock.call_args
self.assertIn('A child node with index', args[0])
def test_next_image_from_file(self, mock):
factory = ResnetImageFromPillowImage()
path = "point_to_somewhere"
factory.next_image_from_file(path)
mock.assert_called_once()
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 testCallError(self, mock):
node = base.NodeBase(None)
node(0)
mock.assert_called_once()
args, _ = mock.call_args
self.assertIn('A child node with index', args[0])
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()
