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