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)