class FactualAPITestSuite(unittest.TestCase):
def setUp(self):
self.factual = Factual(KEY, SECRET)
self.places = self.factual.table('places')
self.facets = self.factual.facets('global')
def test_search(self):
q = self.places.search('factual')
row = q.data()[0]
self.assertRegexpMatches(row['name'], 'Factual')
# full text search for "Mcdonald's, Santa Monica" (test spaces, commas, and apostrophes)
def test_search2(self):
q = self.places.search("McDonald's,Santa Monica").limit(20)
included_rows = q.included_rows()
self.assertEqual(20, included_rows)
def test_limit(self):
q = self.places.search('sushi').limit(3)
self.assertEqual(3, len(q.data()))
def test_select(self):
q = self.places.select('name,address')
row = q.data()[0]
self.assertEqual(2, len(row.keys()))
def test_sort(self):
q1 = self.places.sort_asc('name')
self.assertTrue(q1.data()[0]['name'] < q1.data()[1]['name'])
q2 = self.places.sort_desc('name')
self.assertTrue(q2.data()[0]['name'] > q2.data()[1]['name'])
def test_paging(self):
q1 = self.places.offset(30)
r1 = q1.data()[0]
q2 = self.places.page(3, 15)
r2 = q2.data()[0]
self.assertEqual(r1['name'], r2['name'])
def test_filters(self):
q = self.places.filters({'region': 'NV'})
for r in q.data():
self.assertEqual('NV', r['region'])
def test_geo(self):
q = self.places.search('factual').geo(circle(34.06021, -118.41828, 1000))
row = q.data()[0]
self.assertEqual('Factual', row['name'])
self.assertEqual('1801 Avenue Of The Stars', row['address'])
def test_read_user(self):
q = self.places.search('sushi').user('python_driver_tester')
included_rows = q.included_rows()
self.assertEqual(20, included_rows)
def test_resolve(self):
q = self.factual.resolve('places-v3', {'name': 'factual inc', 'locality': 'los angeles'})
row = q.data()[0]
self.assertTrue(row['resolved'])
self.assertEqual('1801 Avenue Of The Stars', row['address'])
def test_crosswalk(self):
q = self.factual.crosswalk()
result = q.filters({'factual_id':'03c26917-5d66-4de9-96bc-b13066173c65','namespace':'simplegeo'}).data()
self.assertEqual(1, len(result))
self.assertEqual('SG_3ueEyOH4YN3ob9ryHjV1ey', result[0]['namespace_id'])
def test_schema(self):
schema = self.places.schema()
self.assertGreater(len(schema['fields']),20)
self.assertIn('title', schema)
self.assertIn('locality', set(f['name'] for f in schema['fields']))
# full text search for things where locality equals 大阪市 (Osaka: test unicode)
def test_unicode(self):
q = self.places.filters({'locality': u'大阪市'})
for r in q.data():
self.assertEqual(u'大阪市', r['locality'])
def test_bw_encoding(self):
q = self.places.filters({'name': {"$bw":"Star"}})
row = q.data()[0]
self.assertRegexpMatches(row['name'], "Star")
def test_in(self):
q = self.places.filters({"locality":{"$in":["Santa Monica","Los Angeles","Culver City"]}})
row = q.data()[0]
self.assertIn(row['locality'], ["Santa Monica","Los Angeles","Culver City"])
def test_and(self):
q = self.places.filters({"$and":[{"country":"US"},{"website":{"$blank":False}}]})
row = q.data()[0]
self.assertEqual('US', row['country'].upper())
self.assertRegexpMatches(row['website'], 'http')
def test_includes(self):
q = self.places.filters({'category_ids':{'$includes':10}})
rows = q.data()
for row in rows:
self.assertIn(10, row['category_ids'])
def test_raw_read(self):
# filters here is url encoded {"name":"Starbucks"}
response = self.factual.raw_read('t/places/read', 'limit=15&filters=%7B%22name%22%3A%22Starbucks%22%7D')
payload = json.loads(response)
data = payload['response']['data']
self.assertEqual(15, payload['response']['included_rows'])
self.assertTrue(all(row['name'] == 'Starbucks' for row in data))
def test_raw_read_with_map(self):
response = self.factual.raw_read('t/places/read', {'limit':15,'filters':{"name":"Starbucks"}})
payload = json.loads(response)
data = payload['response']['data']
self.assertEqual(15, payload['response']['included_rows'])
self.assertTrue(all(row['name'] == 'Starbucks' for row in data))
def test_raw_write(self):
uuid = '1007462b-dd79-44f5-a69f-e0b6041fa8bd'
params = {'problem':'other','user':'******','debug':True}
response = self.factual.raw_write('t/us-sandbox/' + uuid + '/flag', params)
payload = json.loads(response)
self.assertEqual('ok', payload['status'])
def test_facets1(self):
q = self.facets.search("starbucks").select("country")
results = q.data()['country']
self.assertTrue(results['us'] > 5000)
self.assertTrue(results['ca'] > 200)
def test_facets2(self):
q = self.facets.search("starbucks").select("locality,region").filters({"country":"US"})
locality = q.data()['locality']
region = q.data()['region']
self.assertTrue(locality['chicago'] > 50)
self.assertTrue(region['tx'] > 200)
def test_facets_geo(self):
q = self.facets.select("locality").geo(circle(34.06018, -118.41835, 1000))
locality = q.data()['locality']
self.assertTrue(locality['los angeles'] > 3000)
self.assertTrue(locality['beverly hills'] > 500)
def test_geopulse(self):
geopulse = self.factual.geopulse(point(34.06021, -118.41828))
income_only = geopulse.select('income')
all_results = geopulse.data()
income_results = income_only.data()
self.assertIn('demographics', all_results)
demographics = all_results['demographics']
self.assertIn('area_statistics', demographics)
area_statistics = demographics['area_statistics']
self.assertIn('commercial_residential', area_statistics)
commercial_residential = area_statistics['commercial_residential']
self.assertTrue(0 <= commercial_residential['business'] <= 100)
income_demographics = income_results['demographics']
self.assertEqual(1, len(income_demographics))
self.assertIn('income', income_demographics)
def test_geocode(self):
geocode = self.factual.geocode(point(34.06021, -118.41828))
result = geocode.data()[0]
self.assertEqual('1801 Avenue Of The Stars', result['address'])
self.assertLess(result['$distance'], 20)
def test_monetize(self):
monetize = self.factual.monetize().filters({'place_locality':'Los Angeles'})
result = monetize.data()
self.assertGreaterEqual(len(result), 1)
self.assertTrue(all(row['place_locality'] == 'Los Angeles' for row in result))
def test_submit_without_id(self):
values = {'longitude': 100}
submit = self.factual.submit('us-sandbox', values=values).user('python_driver_tester')
response = submit.write()
if 'new_entity' in response:
self.assertTrue(response['new_entity'])
else:
self.assertIn('status', response)
self.assertEqual('warning', response['status'])
def test_submit_with_id(self):
values = {'longitude': 100}
submit = self.factual.submit('us-sandbox', factual_id='1007462b-dd79-44f5-a69f-e0b6041fa8bd', values=values).user('python_driver_tester')
response = submit.write()
if 'new_entity' in response:
self.assertFalse(response['new_entity'])
else:
self.assertIn('status', response)
self.assertEqual('warning', response['status'])
def test_clear(self):
clear = self.factual.clear('us-sandbox', '1007462b-dd79-44f5-a69f-e0b6041fa8bd', 'latitude,longitude').user('python_driver_tester')
response = clear.write()
self.assertIn('commit_id', response)
self.assertGreater(len(response['commit_id']), 0)
def test_flag(self):
flag = self.factual.flag('us-sandbox', '1007462b-dd79-44f5-a69f-e0b6041fa8bd').user('python_driver_tester').other().debug(True)
response = flag.write()
self.assertEqual('ok', response['status'])
def test_diffs(self):
diffs = self.factual.diffs('2EH4Pz', 1339123455775, 1339136968687).data()
self.assertGreater(len(diffs), 0)
def test_diffs_streaming(self):
diff_request = self.factual.diffs('2EH4Pz', 1339123455775, 1339136968687)
batch = diff_request.data()
streamed = list(diff_request.stream())
self.assertItemsEqual(batch, streamed)
def test_match(self):
match = self.factual.match('places-v3', {'name':'McDonalds','address':'10451 Santa Monica Blvd','locality':'Los Angeles','region':'CA'})
match_id = match.get_id()
self.assertEqual('c730d193-ba4d-4e98-8620-29c672f2f117', match_id)
def test_multi(self):
q1 = self.factual.table('places').filters({'postcode':'90067'})
q2 = self.factual.facets('places').filters({'postcode':'90067'}).select('category_labels')
response = self.factual.multi({'query1':q1,'query2':q2})
self.assertTrue('query1' in response and 'query2' in response)
def test_get_row(self):
row = self.factual.get_row('places', '03c26917-5d66-4de9-96bc-b13066173c65')
self.assertEqual('Factual', row['name'])
class FactualAPITestSuite(unittest.TestCase):
def setUp(self):
self.factual = Factual(KEY, SECRET)
self.places = self.factual.table('places')
self.facets = self.factual.facets('global')
def test_search(self):
q = self.places.search('factual')
row = q.data()[0]
self.assertRegexpMatches(row['name'], 'Factual')
# full text search for "Mcdonald's, Santa Monica" (test spaces, commas, and apostrophes)
def test_search2(self):
q = self.places.search("McDonald's,Santa Monica").limit(20)
included_rows = q.included_rows()
self.assertEqual(20, included_rows)
def test_limit(self):
q = self.places.search('sushi').limit(3)
self.assertEqual(3, len(q.data()))
def test_select(self):
q = self.places.select('name,address')
row = q.data()[0]
self.assertEqual(2, len(row.keys()))
def test_sort(self):
q1 = self.places.sort_asc('name')
self.assertTrue(q1.data()[0]['name'] < q1.data()[1]['name'])
q2 = self.places.sort_desc('name')
self.assertTrue(q2.data()[0]['name'] > q2.data()[1]['name'])
def test_paging(self):
q1 = self.places.offset(30)
r1 = q1.data()[0]
q2 = self.places.page(3, 15)
r2 = q2.data()[0]
self.assertEqual(r1['name'], r2['name'])
def test_filters(self):
q = self.places.filters({'region': 'NV'})
for r in q.data():
self.assertEqual('NV', r['region'])
def test_geo(self):
q = self.places.search('factual').geo(circle(34.06021, -118.41828, 1000))
row = q.data()[0]
self.assertEqual('Factual', row['name'])
self.assertEqual('1801 Avenue Of The Stars', row['address'])
def test_resolve(self):
q = self.factual.resolve({'name': 'factual inc', 'locality': 'los angeles'})
row = q.data()[0]
self.assertTrue(row['resolved'])
self.assertEqual('1801 Avenue Of The Stars', row['address'])
def test_crosswalk(self):
q = self.factual.crosswalk()
result = q.filters({'factual_id':'03c26917-5d66-4de9-96bc-b13066173c65','namespace':'simplegeo'}).data()
self.assertEqual(1, len(result))
self.assertEqual('SG_3ueEyOH4YN3ob9ryHjV1ey', result[0]['namespace_id'])
def test_schema(self):
schema = self.places.schema()
self.assertEqual(21, len(schema['fields']))
self.assertIn('title', schema)
self.assertIn('locality', set(f['name'] for f in schema['fields']))
# full text search for things where locality equals 大阪市 (Osaka: test unicode)
def test_unicode(self):
q = self.places.filters({'locality': '大阪市'})
for r in q.data():
self.assertEqual('大阪市', r['locality'])
def test_bw_encoding(self):
q = self.places.filters({'category': {"$bw":"Arts, Entertainment & Nightlife > Bars"}})
row = q.data()[0]
self.assertRegexpMatches(row['category'], "Arts, Entertainment & Nightlife > Bars")
def test_in(self):
q = self.places.filters({"locality":{"$in":["Santa Monica","Los Angeles","Culver City"]}})
row = q.data()[0]
self.assertIn(row['locality'], ["Santa Monica","Los Angeles","Culver City"])
def test_and(self):
q = self.places.filters({"$and":[{"country":"US"},{"website":{"$blank":False}}]})
row = q.data()[0]
self.assertEqual('US', row['country'])
self.assertRegexpMatches(row['website'], 'http')
def test_raw_read(self):
# filters here is url encoded {"name":"Starbucks"}
response = self.factual.raw_read('t/places/read', 'limit=15&filters=%7B%22name%22%3A%22Starbucks%22%7D')
payload = json.loads(response)
data = payload['response']['data']
self.assertEqual(15, payload['response']['included_rows'])
self.assertTrue(all(row['name'] == 'Starbucks' for row in data))
def test_raw_read_with_map(self):
response = self.factual.raw_read('t/places/read', {'limit':15,'filters':{"name":"Starbucks"}})
payload = json.loads(response)
data = payload['response']['data']
self.assertEqual(15, payload['response']['included_rows'])
self.assertTrue(all(row['name'] == 'Starbucks' for row in data))
def test_facets1(self):
q = self.facets.search("starbucks").select("country")
results = q.data()['country']
self.assertTrue(results['us'] > 5000)
self.assertTrue(results['ca'] > 200)
def test_facets2(self):
q = self.facets.search("starbucks").select("locality,region").filters({"country":"US"})
locality = q.data()['locality']
region = q.data()['region']
self.assertTrue(locality['chicago'] > 50)
self.assertTrue(region['tx'] > 200)
def test_facets_geo(self):
q = self.facets.select("category").geo(circle(34.06018, -118.41835, 5000))
category = q.data()['category']
self.assertTrue(category['shopping'] > 1000)
self.assertTrue(category['health & medicine > physicians'] > 1000)
def test_geopulse(self):
geopulse = self.factual.geopulse(point(34.06021, -118.41828))
income_only = geopulse.select('income')
all_results = geopulse.data()[0]
income_results = income_only.data()[0]
self.assertIn('commercial_density', all_results)
self.assertTrue(0 <= all_results['commercial_density'] <= 1)
self.assertEqual(1, len(income_results))
self.assertIn('income', income_results)
def test_geocode(self):
geocode = self.factual.geocode(point(34.06021, -118.41828))
result = geocode.data()[0]
self.assertEqual('1801 Avenue Of The Stars', result['address'])
self.assertLess(result['$distance'], 20)
def test_monetize(self):
monetize = self.factual.monetize().filters({'place_locality':'Los Angeles'})
result = monetize.data()
self.assertGreaterEqual(len(result), 1)
self.assertTrue(all(row['place_locality'] == 'Los Angeles' for row in result))
def test_submit_without_id(self):
values = {'longitude': 100}
submit = self.factual.submit('t7RSEV', values=values).user('python_driver_tester')
response = submit.write()
if 'new_entity' in response:
self.assertTrue(response['new_entity'])
else:
self.assertIn('status', response)
self.assertEqual('warning', response['status'])
def test_submit_with_id(self):
values = {'longitude': 100}
submit = self.factual.submit('t7RSEV', factual_id='bf6547a8-c0f3-4ada-abf8-9b831bbd5eeb', values=values).user('python_driver_tester')
response = submit.write()
if 'new_entity' in response:
self.assertFalse(response['new_entity'])
else:
self.assertIn('status', response)
self.assertEqual('warning', response['status'])
def test_flag(self):
flag = self.factual.flag('t7RSEV', 'bf6547a8-c0f3-4ada-abf8-9b831bbd5eeb').user('python_driver_tester').other().debug(True)
response = flag.write()
self.assertEqual('ok', response['status'])
def test_diffs(self):
diffs = self.factual.diffs('2EH4Pz', 1339123455775, 1339136968687).data()
self.assertGreater(len(diffs), 0)
def test_diffs_streaming(self):
diff_request = self.factual.diffs('2EH4Pz', 1339123455775, 1339136968687)
batch = diff_request.data()
streamed = list(diff_request.stream())
self.assertItemsEqual(batch, streamed)
def test_match(self):
match = self.factual.match({'name':'McDonalds','address':'10451 Santa Monica Blvd','locality':'Los Angeles','region':'CA'})
match_id = match.get_id()
self.assertEqual('bd886f67-9d86-40c5-9217-f7bcd53cfc0e', match_id)
def test_multi(self):
q1 = self.factual.table('places').filters({'postcode':'90067'})
q2 = self.factual.facets('places').filters({'postcode':'90067'}).select('category')
response = self.factual.multi({'query1':q1,'query2':q2})
self.assertTrue('query1' in response and 'query2' in response)
class FactualAPITestSuite(unittest.TestCase):
def setUp(self):
self.factual = Factual(KEY, SECRET)
self.places = self.factual.table('places')
self.facets = self.factual.facets('global')
def test_search(self):
q = self.places.search('factual')
row = q.data()[0]
self.assertRegexpMatches(row['name'], 'Factual')
# full text search for "Mcdonald's, Santa Monica" (test spaces, commas, and apostrophes)
def test_search2(self):
q = self.places.search("McDonald's,Santa Monica").limit(20)
included_rows = q.included_rows()
self.assertEqual(20, included_rows)
def test_limit(self):
q = self.places.search('sushi').limit(3)
self.assertEqual(3, len(q.data()))
def test_select(self):
q = self.places.select('name,address')
row = q.data()[0]
self.assertEqual(2, len(row.keys()))
def test_sort(self):
q1 = self.places.sort_asc('name')
self.assertTrue(q1.data()[0]['name'] < q1.data()[1]['name'])
q2 = self.places.sort_desc('name')
self.assertTrue(q2.data()[0]['name'] > q2.data()[1]['name'])
def test_paging(self):
q1 = self.places.offset(30)
r1 = q1.data()[0]
q2 = self.places.page(3, 15)
r2 = q2.data()[0]
self.assertEqual(r1['name'], r2['name'])
def test_filters(self):
q = self.places.filters({'region': 'NV'})
for r in q.data():
self.assertEqual('NV', r['region'])
def test_geo(self):
q = self.places.search('factual').geo(
circle(34.06021, -118.41828, 1000))
row = q.data()[0]
self.assertEqual('Factual', row['name'])
self.assertEqual('1801 Avenue Of The Stars', row['address'])
def test_read_user(self):
q = self.places.search('sushi').user('python_driver_tester')
included_rows = q.included_rows()
self.assertEqual(20, included_rows)
def test_resolve(self):
q = self.factual.resolve('places-v3', {
'name': 'factual inc',
'locality': 'los angeles'
})
row = q.data()[0]
self.assertTrue(row['resolved'])
self.assertEqual('1801 Avenue Of The Stars', row['address'])
def test_crosswalk(self):
q = self.factual.crosswalk()
result = q.filters({
'factual_id': '03c26917-5d66-4de9-96bc-b13066173c65',
'namespace': 'simplegeo'
}).data()
self.assertEqual(1, len(result))
self.assertEqual('SG_3ueEyOH4YN3ob9ryHjV1ey',
result[0]['namespace_id'])
def test_schema(self):
schema = self.places.schema()
self.assertGreater(len(schema['fields']), 20)
self.assertIn('title', schema)
self.assertIn('locality', set(f['name'] for f in schema['fields']))
# full text search for things where locality equals 大阪市 (Osaka: test unicode)
def test_unicode(self):
q = self.places.filters({'locality': u'大阪市'})
for r in q.data():
self.assertEqual(u'大阪市', r['locality'])
def test_bw_encoding(self):
q = self.places.filters({'name': {"$bw": "Star"}})
row = q.data()[0]
self.assertRegexpMatches(row['name'], "Star")
def test_in(self):
q = self.places.filters({
"locality": {
"$in": ["Santa Monica", "Los Angeles", "Culver City"]
}
})
row = q.data()[0]
self.assertIn(row['locality'],
["Santa Monica", "Los Angeles", "Culver City"])
def test_and(self):
q = self.places.filters(
{"$and": [{
"country": "US"
}, {
"website": {
"$blank": False
}
}]})
row = q.data()[0]
self.assertEqual('US', row['country'].upper())
self.assertRegexpMatches(row['website'], 'http')
def test_includes(self):
q = self.places.filters({'category_ids': {'$includes': 10}})
rows = q.data()
for row in rows:
self.assertIn(10, row['category_ids'])
def test_raw_read(self):
# filters here is url encoded {"name":"Starbucks"}
response = self.factual.raw_read(
't/places/read',
'limit=15&filters=%7B%22name%22%3A%22Starbucks%22%7D')
payload = json.loads(response)
data = payload['response']['data']
self.assertEqual(15, payload['response']['included_rows'])
self.assertTrue(all(row['name'] == 'Starbucks' for row in data))
def test_raw_read_with_map(self):
response = self.factual.raw_read('t/places/read', {
'limit': 15,
'filters': {
"name": "Starbucks"
}
})
payload = json.loads(response)
data = payload['response']['data']
self.assertEqual(15, payload['response']['included_rows'])
self.assertTrue(all(row['name'] == 'Starbucks' for row in data))
def test_raw_write(self):
uuid = '1007462b-dd79-44f5-a69f-e0b6041fa8bd'
params = {
'problem': 'other',
'user': '******',
'debug': True
}
response = self.factual.raw_write('t/us-sandbox/' + uuid + '/flag',
params)
payload = json.loads(response)
self.assertEqual('ok', payload['status'])
def test_facets1(self):
q = self.facets.search("starbucks").select("country")
results = q.data()['country']
self.assertTrue(results['us'] > 5000)
self.assertTrue(results['ca'] > 200)
def test_facets2(self):
q = self.facets.search("starbucks").select("locality,region").filters(
{"country": "US"})
locality = q.data()['locality']
region = q.data()['region']
self.assertTrue(locality['chicago'] > 50)
self.assertTrue(region['tx'] > 200)
def test_facets_geo(self):
q = self.facets.select("locality").geo(
circle(34.06018, -118.41835, 1000))
locality = q.data()['locality']
self.assertTrue(locality['los angeles'] > 3000)
self.assertTrue(locality['beverly hills'] > 500)
def test_geopulse(self):
geopulse = self.factual.geopulse(point(34.06021, -118.41828))
income_only = geopulse.select('income')
all_results = geopulse.data()
income_results = income_only.data()
self.assertIn('demographics', all_results)
demographics = all_results['demographics']
self.assertIn('area_statistics', demographics)
area_statistics = demographics['area_statistics']
self.assertIn('commercial_residential', area_statistics)
commercial_residential = area_statistics['commercial_residential']
self.assertTrue(0 <= commercial_residential['business'] <= 100)
income_demographics = income_results['demographics']
self.assertEqual(1, len(income_demographics))
self.assertIn('income', income_demographics)
def test_geocode(self):
geocode = self.factual.geocode(point(34.06021, -118.41828))
result = geocode.data()[0]
self.assertEqual('1801 Avenue Of The Stars', result['address'])
self.assertLess(result['$distance'], 20)
def test_monetize(self):
monetize = self.factual.monetize().filters(
{'place_locality': 'Los Angeles'})
result = monetize.data()
self.assertGreaterEqual(len(result), 1)
self.assertTrue(
all(row['place_locality'] == 'Los Angeles' for row in result))
def test_submit_without_id(self):
values = {'longitude': 100}
submit = self.factual.submit(
'us-sandbox', values=values).user('python_driver_tester')
response = submit.write()
if 'new_entity' in response:
self.assertTrue(response['new_entity'])
else:
self.assertIn('status', response)
self.assertEqual('warning', response['status'])
def test_submit_with_id(self):
values = {'longitude': 100}
submit = self.factual.submit(
'us-sandbox',
factual_id='1007462b-dd79-44f5-a69f-e0b6041fa8bd',
values=values).user('python_driver_tester')
response = submit.write()
if 'new_entity' in response:
self.assertFalse(response['new_entity'])
else:
self.assertIn('status', response)
self.assertEqual('warning', response['status'])
def test_clear(self):
clear = self.factual.clear(
'us-sandbox', '1007462b-dd79-44f5-a69f-e0b6041fa8bd',
'latitude,longitude').user('python_driver_tester')
response = clear.write()
self.assertIn('commit_id', response)
self.assertGreater(len(response['commit_id']), 0)
def test_flag(self):
flag = self.factual.flag(
'us-sandbox', '1007462b-dd79-44f5-a69f-e0b6041fa8bd').user(
'python_driver_tester').other().debug(True)
response = flag.write()
self.assertEqual('ok', response['status'])
def test_diffs(self):
diffs = self.factual.diffs('2EH4Pz', 1339123455775,
1339136968687).data()
self.assertGreater(len(diffs), 0)
def test_diffs_streaming(self):
diff_request = self.factual.diffs('2EH4Pz', 1339123455775,
1339136968687)
batch = diff_request.data()
streamed = list(diff_request.stream())
self.assertItemsEqual(batch, streamed)
def test_match(self):
match = self.factual.match(
'places-v3', {
'name': 'McDonalds',
'address': '10451 Santa Monica Blvd',
'locality': 'Los Angeles',
'region': 'CA'
})
match_id = match.get_id()
self.assertEqual('c730d193-ba4d-4e98-8620-29c672f2f117', match_id)
def test_multi(self):
q1 = self.factual.table('places').filters({'postcode': '90067'})
q2 = self.factual.facets('places').filters({
'postcode': '90067'
}).select('category_labels')
response = self.factual.multi({'query1': q1, 'query2': q2})
self.assertTrue('query1' in response and 'query2' in response)
def test_get_row(self):
row = self.factual.get_row('places',
'03c26917-5d66-4de9-96bc-b13066173c65')
self.assertEqual('Factual', row['name'])
class FactualAPITestSuite(unittest.TestCase):
def setUp(self):
self.factual = Factual(KEY, SECRET)
self.places = self.factual.table('places')
self.facets = self.factual.facets('global')
def test_search(self):
q = self.places.search('factual')
row = q.data()[0]
self.assertRegexpMatches(row['name'], 'Factual')
# full text search for "Mcdonald's, Santa Monica" (test spaces, commas, and apostrophes)
def test_search2(self):
q = self.places.search("McDonald's,Santa Monica").limit(20)
included_rows = q.included_rows()
self.assertEqual(20, included_rows)
def test_limit(self):
q = self.places.search('sushi').limit(3)
self.assertEqual(3, len(q.data()))
def test_select(self):
q = self.places.select('name,address')
row = q.data()[0]
self.assertEqual(2, len(row.keys()))
def test_sort(self):
q1 = self.places.sort_asc('name')
self.assertTrue(q1.data()[0]['name'] < q1.data()[1]['name'])
q2 = self.places.sort_desc('name')
self.assertTrue(q2.data()[0]['name'] > q2.data()[1]['name'])
def test_paging(self):
q1 = self.places.offset(30)
r1 = q1.data()[0]
q2 = self.places.page(3, 15)
r2 = q2.data()[0]
self.assertEqual(r1['name'], r2['name'])
def test_filters(self):
q = self.places.filters({'region': 'NV'})
for r in q.data():
self.assertEqual('NV', r['region'])
def test_geo(self):
q = self.places.search('factual').geo(
circle(34.06021, -118.41828, 1000))
row = q.data()[0]
self.assertEqual('Factual', row['name'])
self.assertEqual('1801 Avenue Of The Stars', row['address'])
def test_resolve(self):
q = self.factual.resolve({
'name': 'factual inc',
'locality': 'los angeles'
})
row = q.data()[0]
self.assertTrue(row['resolved'])
self.assertEqual('1801 Avenue Of The Stars', row['address'])
def test_crosswalk(self):
q = self.factual.crosswalk()
result = q.filters({
'factual_id': '03c26917-5d66-4de9-96bc-b13066173c65',
'namespace': 'simplegeo'
}).data()
self.assertEqual(1, len(result))
self.assertEqual('SG_3ueEyOH4YN3ob9ryHjV1ey',
result[0]['namespace_id'])
def test_schema(self):
schema = self.places.schema()
self.assertEqual(21, len(schema['fields']))
self.assertIn('title', schema)
self.assertIn('locality', set(f['name'] for f in schema['fields']))
# full text search for things where locality equals 大阪市 (Osaka: test unicode)
def test_unicode(self):
q = self.places.filters({'locality': '大阪市'})
for r in q.data():
self.assertEqual('大阪市', r['locality'])
def test_bw_encoding(self):
q = self.places.filters(
{'category': {
"$bw": "Arts, Entertainment & Nightlife > Bars"
}})
row = q.data()[0]
self.assertRegexpMatches(row['category'],
"Arts, Entertainment & Nightlife > Bars")
def test_in(self):
q = self.places.filters({
"locality": {
"$in": ["Santa Monica", "Los Angeles", "Culver City"]
}
})
row = q.data()[0]
self.assertIn(row['locality'],
["Santa Monica", "Los Angeles", "Culver City"])
def test_and(self):
q = self.places.filters(
{"$and": [{
"country": "US"
}, {
"website": {
"$blank": False
}
}]})
row = q.data()[0]
self.assertEqual('US', row['country'])
self.assertRegexpMatches(row['website'], 'http')
def test_raw_read(self):
# filters here is url encoded {"name":"Starbucks"}
response = self.factual.raw_read(
't/places/read',
'limit=15&filters=%7B%22name%22%3A%22Starbucks%22%7D')
payload = json.loads(response)
data = payload['response']['data']
self.assertEqual(15, payload['response']['included_rows'])
self.assertTrue(all(row['name'] == 'Starbucks' for row in data))
def test_raw_read_with_map(self):
response = self.factual.raw_read('t/places/read', {
'limit': 15,
'filters': {
"name": "Starbucks"
}
})
payload = json.loads(response)
data = payload['response']['data']
self.assertEqual(15, payload['response']['included_rows'])
self.assertTrue(all(row['name'] == 'Starbucks' for row in data))
def test_facets1(self):
q = self.facets.search("starbucks").select("country")
results = q.data()['country']
self.assertTrue(results['us'] > 5000)
self.assertTrue(results['ca'] > 200)
def test_facets2(self):
q = self.facets.search("starbucks").select("locality,region").filters(
{"country": "US"})
locality = q.data()['locality']
region = q.data()['region']
self.assertTrue(locality['chicago'] > 50)
self.assertTrue(region['tx'] > 200)
def test_facets_geo(self):
q = self.facets.select("category").geo(
circle(34.06018, -118.41835, 5000))
category = q.data()['category']
self.assertTrue(category['shopping'] > 1000)
self.assertTrue(category['health & medicine > physicians'] > 1000)
def test_geopulse(self):
geopulse = self.factual.geopulse(point(34.06021, -118.41828))
income_only = geopulse.select('income')
all_results = geopulse.data()[0]
income_results = income_only.data()[0]
self.assertIn('commercial_density', all_results)
self.assertTrue(0 <= all_results['commercial_density'] <= 1)
self.assertEqual(1, len(income_results))
self.assertIn('income', income_results)
def test_geocode(self):
geocode = self.factual.geocode(point(34.06021, -118.41828))
result = geocode.data()[0]
self.assertEqual('1801 Avenue Of The Stars', result['address'])
self.assertLess(result['$distance'], 20)
def test_monetize(self):
monetize = self.factual.monetize().filters(
{'place_locality': 'Los Angeles'})
result = monetize.data()
self.assertGreaterEqual(len(result), 1)
self.assertTrue(
all(row['place_locality'] == 'Los Angeles' for row in result))
def test_submit_without_id(self):
values = {'longitude': 100}
submit = self.factual.submit(
't7RSEV', values=values).user('python_driver_tester')
response = submit.write()
if 'new_entity' in response:
self.assertTrue(response['new_entity'])
else:
self.assertIn('status', response)
self.assertEqual('warning', response['status'])
def test_submit_with_id(self):
values = {'longitude': 100}
submit = self.factual.submit(
't7RSEV',
factual_id='bf6547a8-c0f3-4ada-abf8-9b831bbd5eeb',
values=values).user('python_driver_tester')
response = submit.write()
if 'new_entity' in response:
self.assertFalse(response['new_entity'])
else:
self.assertIn('status', response)
self.assertEqual('warning', response['status'])
def test_flag(self):
flag = self.factual.flag(
't7RSEV', 'bf6547a8-c0f3-4ada-abf8-9b831bbd5eeb').user(
'python_driver_tester').other().debug(True)
response = flag.write()
self.assertEqual('ok', response['status'])
def test_diffs(self):
diffs = self.factual.diffs('2EH4Pz', 1339123455775,
1339136968687).data()
self.assertGreater(len(diffs), 0)
def test_diffs_streaming(self):
diff_request = self.factual.diffs('2EH4Pz', 1339123455775,
1339136968687)
batch = diff_request.data()
streamed = list(diff_request.stream())
self.assertItemsEqual(batch, streamed)
def test_match(self):
match = self.factual.match({
'name': 'McDonalds',
'address': '10451 Santa Monica Blvd',
'locality': 'Los Angeles',
'region': 'CA'
})
match_id = match.get_id()
self.assertEqual('bd886f67-9d86-40c5-9217-f7bcd53cfc0e', match_id)
def test_multi(self):
q1 = self.factual.table('places').filters({'postcode': '90067'})
q2 = self.factual.facets('places').filters({
'postcode': '90067'
}).select('category')
response = self.factual.multi({'query1': q1, 'query2': q2})
self.assertTrue('query1' in response and 'query2' in response)