dataset: Dataset) -> Tuple[DatasetReader, ArtifactCollection]: return TestDatasetReader(), Blobs( {'data': InMemoryBlob(dataset.data.encode('utf8'))}) class TestDatasetReader(DatasetReader): def read(self, artifacts: ArtifactCollection) -> Dataset: return Dataset(artifacts.bytes_dict()['data'].decode('utf8'), TestDatasetType()) class TestDatasetType(DatasetType): def get_writer(self): return TestDatasetWriter() def deserialize(self, obj: dict) -> object: return obj def serialize(self, instance: object) -> dict: return instance @pytest.fixture def data() -> Dataset: return Dataset('abcdefg', TestDatasetType()) create_dataset_hooks = interface_hook_creator('tests/repository/dataset/', 'dataset_common.py', 'dataset_repo')
image.model = created_model return meta.create_image(image) @pytest.fixture def environment(): return RuntimeEnvironment("Test Environment", params=TestParams(123)) @pytest.fixture def created_environment(meta, environment): environment = deepcopy(environment) return meta.create_environment(environment) @pytest.fixture def instance(): return RuntimeInstance("Test Instance", params=TestParams(123)) @pytest.fixture def created_instance(meta, created_image, created_environment, instance): instance = deepcopy(instance) instance.image = created_image instance.environment = created_environment return meta.create_instance(instance) create_metadata_hooks = interface_hook_creator('tests/repository/metadata/', 'meta_common.py', 'meta')
image.params.name = 'image' image.bind_builder(mock_env.params.get_builder()) image = ebnt.meta_repo.create_image(image) yield image @pytest.fixture def regression_and_data(): reg = LinearRegression() data = pd.DataFrame([[1, 1], [2, 1]], columns=['a', 'b']) reg.fit(data, [1, 0]) return reg, data @pytest.fixture def image(): return Image('Test Image', BuildableMock()) @pytest.fixture def pipeline(model): return model.as_pipeline() @pytest.fixture def instance(): return RuntimeInstance('Test Instance') create_client_hooks = interface_hook_creator('tests/client/', 'client_common.py', 'ebnt')
import pytest from ebonite.core.objects.artifacts import InMemoryBlob from ebonite.core.objects.core import Model from tests.conftest import interface_hook_creator @pytest.fixture def model(): mdl = Model.create(lambda data: data, 'input', 'test_model') mdl._id = 'test_model_id' return mdl @pytest.fixture def blobs(): return {'blob1': InMemoryBlob(b'blob1'), 'blob2': InMemoryBlob(b'blob2')} create_artifact_hooks = interface_hook_creator('tests/repository/artifact/', 'meta_common.py', 'art_repo')