def test1(): datadir = resource_filename('sknano', 'data') dirnames, fnames = listdir(datadir) assert_true('__init__.py' in fnames) assert_true('fullerenes' in dirnames) assert_equal(dirnames, listdir_dirnames(datadir)) assert_equal(fnames, listdir_fnames(datadir))
def load_fullerene_data(): """Helper function to populate dict of fullerene data files.""" datadir = resource_filename('sknano', 'data/fullerenes') fullerenes = \ listdir_dirnames(datadir, filterfunc=lambda name: name.startswith('C')) fullerene_data = {} for fullerene in fullerenes: datapath = os.path.join('data', 'fullerenes', fullerene) datadir = resource_filename('sknano', datapath) fullerene_data[fullerene] = listdir_fnames(datadir) return fullerene_data
def test2(): datadir = resource_filename('sknano', 'data') filterfunc = lambda name: '_' not in name dirnames, fnames = listdir(datadir, filterfunc=filterfunc, filter_fnames=True, filter_dirnames=True) assert_true('__pycache__' not in dirnames) assert_true('__init__.py' not in fnames) assert_equal(dirnames, listdir_dirnames(datadir, filterfunc=filterfunc)) assert_equal(fnames, listdir_fnames(datadir, filterfunc=filterfunc))