def initialize_store(test_data): """Helper function that sets up a default store with some data in it""" global_state.cleanup() global_state.use_default_store() for data_id in ["1", "2"]: global_state.set_data(data_id, test_data) global_state.set_dtypes(data_id, build_dtypes_state(test_data)) global_state.set_settings(data_id, dict(locked=[])) global_state.set_name(data_id, "test_name" + data_id) global_state.set_context_variables( data_id, dict(favorite_words=["foo", "bar", "baz"])) global_state.set_history(data_id, ["foo", "bar", "baz"])
def initialize_store(test_data): """Helper function that sets up a default store with some data in it""" global_state.cleanup() global_state.use_default_store() for data_id in ['1', '2']: global_state.set_data(data_id, test_data) global_state.set_dtypes(data_id, build_dtypes_state(test_data)) global_state.set_settings(data_id, dict(locked=[])) global_state.set_metadata(data_id, dict(name='test_name')) global_state.set_context_variables( data_id, dict(favorite_words=['foo', 'bar', 'baz'])) global_state.set_history(data_id, ['foo', 'bar', 'baz'])
def test_use_default_store(unittest, tmpdir, test_data): """Make sure flipping back and forth multiple times doesn't corrupt the data""" initialize_store(test_data) contents_before = get_store_contents() type_before = get_store_type() directory = tmpdir.mkdir('test_use_default_store').dirname global_state.use_shelve_store(directory) global_state.use_default_store() contents_after = get_store_contents() type_after = get_store_type() unittest.assertEqual(contents_before, contents_after) unittest.assertEqual(type_before, type_after)
def setup_function(function): global_state.cleanup() global_state.use_default_store()
def teardown_function(function): global_state.cleanup() global_state.use_default_store()
def setup_module(module): global_state.cleanup() global_state.use_default_store()
def teardown_module(module): global_state.cleanup() global_state.use_default_store()