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