def test_neptune_additional_methods(neptune): logger = NeptuneLogger(offline_mode=True) logger.log_metric('test', torch.ones(1)) neptune.create_experiment().log_metric.assert_called_once_with( 'test', torch.ones(1)) neptune.create_experiment().log_metric.reset_mock() logger.log_metric('test', 1.0) neptune.create_experiment().log_metric.assert_called_once_with('test', 1.0) neptune.create_experiment().log_metric.reset_mock() logger.log_metric('test', 1.0, step=2) neptune.create_experiment().log_metric.assert_called_once_with('test', x=2, y=1.0) neptune.create_experiment().log_metric.reset_mock() logger.log_text('test', 'text') neptune.create_experiment().log_metric.assert_called_once_with( 'test', 'text') neptune.create_experiment().log_metric.reset_mock() logger.log_image('test', 'image file') neptune.create_experiment().log_image.assert_called_once_with( 'test', 'image file') neptune.create_experiment().log_image.reset_mock() logger.log_image('test', 'image file', step=2) neptune.create_experiment().log_image.assert_called_once_with( 'test', x=2, y='image file') neptune.create_experiment().log_image.reset_mock() logger.log_artifact('file') neptune.create_experiment().log_artifact.assert_called_once_with( 'file', None) logger.set_property('property', 10) neptune.create_experiment().set_property.assert_called_once_with( 'property', 10) logger.append_tags('one tag') neptune.create_experiment().append_tags.assert_called_once_with('one tag') neptune.create_experiment().append_tags.reset_mock() logger.append_tags(['two', 'tags']) neptune.create_experiment().append_tags.assert_called_once_with( 'two', 'tags')
def test_neptune_additional_methods(neptune): logger = NeptuneLogger(api_key='test', project_name='project') created_experiment = neptune.Session.with_default_backend().get_project( ).create_experiment() logger.log_metric('test', torch.ones(1)) created_experiment.log_metric.assert_called_once_with( 'test', torch.ones(1)) created_experiment.log_metric.reset_mock() logger.log_metric('test', 1.0) created_experiment.log_metric.assert_called_once_with('test', 1.0) created_experiment.log_metric.reset_mock() logger.log_metric('test', 1.0, step=2) created_experiment.log_metric.assert_called_once_with('test', x=2, y=1.0) created_experiment.log_metric.reset_mock() logger.log_text('test', 'text') created_experiment.log_text.assert_called_once_with('test', 'text', step=None) created_experiment.log_text.reset_mock() logger.log_image('test', 'image file') created_experiment.log_image.assert_called_once_with('test', 'image file') created_experiment.log_image.reset_mock() logger.log_image('test', 'image file', step=2) created_experiment.log_image.assert_called_once_with('test', x=2, y='image file') created_experiment.log_image.reset_mock() logger.log_artifact('file') created_experiment.log_artifact.assert_called_once_with('file', None) logger.set_property('property', 10) created_experiment.set_property.assert_called_once_with('property', 10) logger.append_tags('one tag') created_experiment.append_tags.assert_called_once_with('one tag') created_experiment.append_tags.reset_mock() logger.append_tags(['two', 'tags']) created_experiment.append_tags.assert_called_once_with('two', 'tags')
def test_neptune_additional_methods(neptune): logger = NeptuneLogger(api_key="test", project_name="project") created_experiment = neptune.Session.with_default_backend().get_project( ).create_experiment() logger.log_metric("test", torch.ones(1)) created_experiment.log_metric.assert_called_once_with( "test", torch.ones(1)) created_experiment.log_metric.reset_mock() logger.log_metric("test", 1.0) created_experiment.log_metric.assert_called_once_with("test", 1.0) created_experiment.log_metric.reset_mock() logger.log_metric("test", 1.0, step=2) created_experiment.log_metric.assert_called_once_with("test", x=2, y=1.0) created_experiment.log_metric.reset_mock() logger.log_text("test", "text") created_experiment.log_text.assert_called_once_with("test", "text") created_experiment.log_text.reset_mock() logger.log_image("test", "image file") created_experiment.log_image.assert_called_once_with("test", "image file") created_experiment.log_image.reset_mock() logger.log_image("test", "image file", step=2) created_experiment.log_image.assert_called_once_with("test", x=2, y="image file") created_experiment.log_image.reset_mock() logger.log_artifact("file") created_experiment.log_artifact.assert_called_once_with("file", None) logger.set_property("property", 10) created_experiment.set_property.assert_called_once_with("property", 10) logger.append_tags("one tag") created_experiment.append_tags.assert_called_once_with("one tag") created_experiment.append_tags.reset_mock() logger.append_tags(["two", "tags"]) created_experiment.append_tags.assert_called_once_with("two", "tags")