# Path: model_card_toolkit/tfx/artifact.py
# class ModelCard(Artifact):
#   TYPE_NAME = 'ModelCard'
#   TYPE_ANNOTATION = Metrics
# def create_and_save_artifact(
#     artifact_name: str, artifact_uri: str,
#     store: mlmd.MetadataStore) -> metadata_store_pb2.Artifact:


from absl.testing import absltest
from model_card_toolkit.tfx import artifact
from ml_metadata.proto import metadata_store_pb2
import ml_metadata as mlmd
"""Tests for artifact."""



class ArtifactTest(absltest.TestCase):

  def setUp(self):
    super(ArtifactTest, self).setUp()
    connection_config = metadata_store_pb2.ConnectionConfig()
    connection_config.fake_database.SetInParent()
    self.store = mlmd.MetadataStore(connection_config)

  def test_create_and_save_artifact(self):

# Path: model_card_toolkit/tfx/artifact.py
# class ModelCard(Artifact):
#   TYPE_NAME = 'ModelCard'
#   TYPE_ANNOTATION = Metrics
# def create_and_save_artifact(
#     artifact_name: str, artifact_uri: str,
#     store: mlmd.MetadataStore) -> metadata_store_pb2.Artifact:


# Path: model_card_toolkit/tfx/artifact_test.py
from absl.testing import absltest
from model_card_toolkit.tfx import artifact
from ml_metadata.proto import metadata_store_pb2
import ml_metadata as mlmd

"""Tests for artifact."""



class ArtifactTest(absltest.TestCase):

  def setUp(self):
    super(ArtifactTest, self).setUp()
    connection_config = metadata_store_pb2.ConnectionConfig()
    connection_config.fake_database.SetInParent()
    self.store = mlmd.MetadataStore(connection_config)

  def test_create_and_save_artifact(self):

    mc_artifact = artifact.create_and_save_artifact(