def check_api_key(self, api_key):
nyt_api = NYT(api_key)
key_valid_flag = nyt_api.check_api_key()
if key_valid_flag:
self.api_key = api_key
self.api_key_updated(key_valid_flag)
def check_api_key(self, api_key):
nyt_api = NYT(api_key)
key_valid_flag = nyt_api.check_api_key()
if key_valid_flag:
self.api_key = api_key
self.api_key_updated(key_valid_flag)
class NYTTests(unittest.TestCase):
API_KEY = 'api_key'
def setUp(self):
self.tmp = tempfile.NamedTemporaryFile(delete=False)
os.remove(self.tmp.name)
self.nyt = NYT(self.API_KEY)
self.nyt.cache_path = self.tmp.name
def tearDown(self):
if os.path.exists(self.tmp.name):
os.remove(self.tmp.name)
def test_nyt_key(self):
self.assertTrue(self.nyt.check_api_key())
def test_nyt_query_keywords(self):
records = self.nyt.run_query('slovenia')
self.assertEqual(len(records), 10)
def test_nyt_query_date_range(self):
corpus = self.nyt.run_query('slovenia', datetime.date(2013, 1, 1),
datetime.date(2014, 1, 1))
self.assertEqual(len(corpus.documents), 10)
def test_nyt_query_max_records(self):
records = self.nyt.run_query('slovenia', max_records=30)
self.assertEqual(len(records), 30)
def test_nyt_corpus_domain_generation(self):
corpus = self.nyt.run_query('slovenia')
meta_vars = [StringVariable.make(field) for field in NYT_TEXT_FIELDS] + \
[StringVariable.make('pub_date'), StringVariable.make('country')]
self.assertEqual(len(meta_vars), len(corpus.domain.metas))
self.assertEqual(len(corpus.Y), 10)
def test_nyt_result_caching(self):
# Run a query to create a cache entry first.
self.nyt.run_query('slovenia')
data, is_cached, error = self.nyt._execute_query(0)
# Check if the response was cached.
self.assertTrue(is_cached)
class NYTTests(unittest.TestCase):
API_KEY = 'api_key'
def setUp(self):
self.tmp = tempfile.NamedTemporaryFile(delete=False)
os.remove(self.tmp.name)
self.nyt = NYT(self.API_KEY)
self.nyt.cache_path = self.tmp.name
def tearDown(self):
if os.path.exists(self.tmp.name):
os.remove(self.tmp.name)
def test_nyt_key(self):
self.assertTrue(self.nyt.check_api_key())
def test_nyt_query_keywords(self):
records = self.nyt.run_query('slovenia')
self.assertEqual(len(records), 10)
def test_nyt_query_date_range(self):
corpus = self.nyt.run_query('slovenia', datetime.date(2013, 1, 1), datetime.date(2014, 1, 1))
self.assertEqual(len(corpus.documents), 10)
def test_nyt_query_max_records(self):
records = self.nyt.run_query('slovenia', max_records=30)
self.assertEqual(len(records), 30)
def test_nyt_corpus_domain_generation(self):
corpus = self.nyt.run_query('slovenia')
meta_vars = [StringVariable.make(field) for field in NYT_TEXT_FIELDS] + \
[StringVariable.make('pub_date'), StringVariable.make('country')]
self.assertEqual(len(meta_vars), len(corpus.domain.metas))
self.assertEqual(len(corpus.Y), 10)
def test_nyt_result_caching(self):
# Run a query to create a cache entry first.
self.nyt.run_query('slovenia')
data, is_cached, error = self.nyt._execute_query(0)
# Check if the response was cached.
self.assertTrue(is_cached)
def check_api_key(self, api_key):
nyt_api = NYT(api_key)
self.api_key = api_key
self.api_key_updated(nyt_api.check_api_key())
def check_api_key(self, api_key):
nyt_api = NYT(api_key)
self.api_key = api_key
self.api_key_updated(nyt_api.check_api_key())
