def test_help_docs(self):
foo_store = OmegaStore(bucket='foo')
reg = LinearRegression()
foo_store.put(reg,
'regmodel',
attributes={'docs': 'this is some text'})
# get backend for different signatures
backend = foo_store._resolve_help_backend('regmodel')
self.assertEqual(backend.__doc__, 'this is some text')
backend = foo_store._resolve_help_backend('regmodel', raw=True)
self.assertIsInstance(backend, ScikitLearnBackend)
def test_help(self):
foo_store = OmegaStore(bucket='foo')
obj = {}
foo_store.put(obj, '_temp')
# get backend for different signatures
backend_name = foo_store._resolve_help_backend('_temp')
backend_obj = foo_store._resolve_help_backend(obj)
self.assertEqual(backend_name, backend_obj)
self.assertIsInstance(backend_obj, OmegaStore)
# get backend for scikit model
reg = LinearRegression()
foo_store.put(reg, 'regmodel')
backend_name = foo_store._resolve_help_backend('regmodel')
backend_obj = foo_store._resolve_help_backend(reg)
self.assertIsInstance(backend_name, ScikitLearnBackend)
self.assertIsInstance(backend_obj, ScikitLearnBackend)
