def test_save_ds_to_file_default_loc(self):
ds = load_default_user_datasource()[0]
filename = save_user_datasource_to(ds)
try:
self.assertEqual(split(filename)[0], get_user_ds_folder())
ds2, filename2 = load_default_user_datasource()
self.assertEqual(filename, filename2)
assert_has_traits_almost_equal(ds, ds2)
finally:
os.remove(filename)
def _data_source_default(self):
from kromatography.utils.app_utils import load_default_user_datasource
return load_default_user_datasource()[0]
def _datasource_default(self):
""" Build a DS from the latest stored version if possible. Build a
default one otherwise.
"""
from kromatography.utils.app_utils import load_default_user_datasource
return load_default_user_datasource()[0]
def test_save_ds_to_custom_file(self):
ds = load_default_user_datasource()[0]
filename = save_user_datasource_to(ds, "test.chromds")
self.assertEqual(filename, "test.chromds")
reloaded_ds, _ = load_object("test.chromds")
assert_has_traits_almost_equal(ds, reloaded_ds)
def test_load_default_user_datasource(self):
ds, filename = load_default_user_datasource()
self.assertIsInstance(ds, SimpleDataSource)
self.assertIn("Prod000", ds.get_object_names_by_type("products"))
self.assertIsInstance(filename, basestring)