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