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