def experiment_metadata(self, experiment_id):
del experiment_id # Unused.
return provider.ExperimentMetadata(
experiment_name="Experiment #5 (実験＃5)",
experiment_description="Take five (😊)",
creation_time=1234.5,
)
def test_attributes(self):
e1 = provider.ExperimentMetadata(
experiment_name="FooExperiment",
experiment_description="Experiment on Foo",
creation_time=1.25,
)
self.assertEqual(e1.experiment_name, "FooExperiment")
self.assertEqual(e1.experiment_description, "Experiment on Foo")
self.assertEqual(e1.creation_time, 1.25)
def test_repr(self):
x = provider.ExperimentMetadata(
experiment_name="FooExperiment",
experiment_description="Experiment on Foo",
creation_time=1.25,
)
repr_ = repr(x)
self.assertIn(repr(x.experiment_name), repr_)
self.assertIn(repr(x.experiment_description), repr_)
self.assertIn(repr(x.creation_time), repr_)
def test_experiment_metadata_when_only_data_location_set(self):
res = data_provider_pb2.GetExperimentResponse()
self.stub.GetExperiment.return_value = res
actual = self.provider.experiment_metadata(self.ctx,
experiment_id="123")
self.assertEqual(actual, provider.ExperimentMetadata())
req = data_provider_pb2.GetExperimentRequest()
req.experiment_id = "123"
self.stub.GetExperiment.assert_called_once_with(req)
def experiment_metadata(self, ctx, *, experiment_id):
req = data_provider_pb2.GetExperimentRequest()
req.experiment_id = experiment_id
with _translate_grpc_error():
res = self._stub.GetExperiment(req)
res = provider.ExperimentMetadata(
data_location=res.data_location,
experiment_name=res.name,
experiment_description=res.description,
creation_time=_timestamp_proto_to_float(res.creation_time),
)
return res
def test_experiment_metadata_with_partial_metadata(self):
res = data_provider_pb2.GetExperimentResponse()
res.name = "mnist"
self.stub.GetExperiment.return_value = res
actual = self.provider.experiment_metadata(self.ctx,
experiment_id="123")
self.assertEqual(
actual,
provider.ExperimentMetadata(
experiment_name="mnist",
experiment_description="",
creation_time=0,
),
)
req = data_provider_pb2.GetExperimentRequest()
req.experiment_id = "123"
self.stub.GetExperiment.assert_called_once_with(req)
def test_experiment_metadata_with_creation_time(self):
res = data_provider_pb2.GetExperimentResponse()
res.name = "mnist"
res.description = "big breakthroughs"
res.creation_time.FromMilliseconds(1500)
self.stub.GetExperiment.return_value = res
actual = self.provider.experiment_metadata(self.ctx,
experiment_id="123")
self.assertEqual(
actual,
provider.ExperimentMetadata(
experiment_name="mnist",
experiment_description="big breakthroughs",
creation_time=1.5,
),
)
req = data_provider_pb2.GetExperimentRequest()
req.experiment_id = "123"
self.stub.GetExperiment.assert_called_once_with(req)
def experiment_metadata(self, ctx, *, experiment_id):
del experiment_id # Unused.
# note: no data location provided
return provider.ExperimentMetadata(experiment_name="name")
def experiment_metadata(self, ctx=None, *, experiment_id):
self._validate_context(ctx)
self._validate_experiment_id(experiment_id)
return provider.ExperimentMetadata(data_location=self._logdir)
def experiment_metadata(self, ctx, *, experiment_id):
self._validate_eid(experiment_id)
data_location = "%s://%s" % (self._name, experiment_id)
return provider.ExperimentMetadata(data_location=data_location)
def md(**kwargs):
kwargs.setdefault("data_location", "/tmp/logs")
kwargs.setdefault("experiment_name", "FooExperiment")
kwargs.setdefault("experiment_description", "Experiment on Foo")
kwargs.setdefault("creation_time", 1.25)
return provider.ExperimentMetadata(**kwargs)
def test_defaults(self):
provider.ExperimentMetadata()
def experiment_metadata(self, ctx, *, experiment_id):
return provider.ExperimentMetadata()
