def test_use_shelve_store(unittest, tmpdir, test_data): initialize_store(test_data) contents_before = get_store_contents() type_before = get_store_type() directory = tmpdir.mkdir("test_use_shelve_store").dirname global_state.use_shelve_store(directory) contents_after = get_store_contents() type_after = get_store_type() unittest.assertEqual(contents_before, contents_after) unittest.assertNotEqual(type_before, type_after) global_state.cleanup(data_id="1") unittest.assertNotEqual(contents_after, get_store_contents())
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)