def test_reset_retriever(tmpdir):
"""Test the dataset reset function."""
pwd_name = os.getcwd()
workdir = tmpdir.mkdtemp()
workdir.chdir()
offline_datasets = rt.dataset_names()['offline']
offline_datasets = [
dataset for dataset in offline_datasets
if not dataset.startswith('test-')
]
if not offline_datasets:
return
dataset = random.choice(offline_datasets)
rt.reset_retriever(dataset)
rt.reload_scripts()
assert os.path.exists(
os.path.join(HOME_DIR,
dataset.replace("-", "_") + ".json")) == False
assert os.path.exists(
os.path.join(HOME_DIR,
dataset.replace("-", "_") + ".py")) == False
if dataset in RETRIEVER_DATASETS:
rt.get_script_upstream(dataset, repo=RETRIEVER_REPOSITORY)
else:
rt.get_script_upstream(dataset)
rt.reload_scripts()
assert dataset in rt.dataset_names()['offline']
os.chdir(pwd_name)
def test_reset_retriever(tmpdir):
"""Test the dataset reset function."""
pwd_name = os.getcwd()
workdir = tmpdir.mkdtemp()
workdir.chdir()
dataset = random.choice(rt.dataset_names())
rt.reset_retriever(dataset)
rt.reload_scripts()
assert dataset not in rt.dataset_names()
rt.check_for_updates()
rt.reload_scripts()
assert dataset in rt.dataset_names()
os.chdir(pwd_name)
def test_dataset_names():
"""Check if dataset names lookup includes a known value"""
assert 'mammal-masses' in rt.dataset_names()
def test_dataset_names():
"""Check if dataset names lookup includes a known value"""
assert 'mammal-masses' in rt.dataset_names()
def test_restiever_test_scripts():
"""Test retriever test scripts"""
assert set(TESTS_SCRIPTS).issubset(set(dataset_names()))
# -*- coding: utf-8 -*-
# Copyright (C) 2019 David Arroyo Menéndez
# Author: David Arroyo Menéndez <*****@*****.**>
# Maintainer: David Arroyo Menéndez <*****@*****.**>
# This file is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3, or (at your option)
# any later version.
# This file is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with GNU Emacs; see the file COPYING. If not, write to
# the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
# Boston, MA 02110-1301 USA,
import retriever as rt
print(rt.dataset_names())
for dataset in rt.datasets():
print(dataset.name)
print(rt.check_for_updates())
print(rt.install_sqlite('iris'))
def test_dataset_names():
"""Check if dataset names lookup includes a known value"""
datasets = rt.dataset_names()
dataset_names = datasets['offline']
dataset_names.extend(datasets['online'])
assert 'mammal-masses' in dataset_names
def test_restiever_test_scripts():
"""Test Retriever test scripts"""
datasets_list = dataset_names()
assert set(TESTS_SCRIPTS).issubset(
set(datasets_list['online'] + datasets_list['offline']))
