def test_query_user_ids_distribution(self):
self.setUp()
args = query_user_id_distribution.parse_args(['-i', os.path.dirname(os.path.abspath(__file__))+'/../test/test_user_ids.json', '-o', os.path.dirname(os.path.abspath(__file__))+'/../test/output.csv'])
query_user_id_distribution.query_distribution(args.output, args.input)
with open(os.path.dirname(os.path.abspath(__file__))+'/../test/test_user_ids.json') as f:
original_total_count = sum(1 for _ in f)
with open(os.path.dirname(os.path.abspath(__file__))+'/../test/output.csv') as f:
reader = csv.DictReader(f)
for row in reader:
if row['id'] == 'all_users_total':
new_total_count = int(row['smapp_tweet_count'])
self.assertEqual(original_total_count,new_total_count)
self.tearDown()
def test_query_user_ids_distribution(self):
self.setUp()
args = query_user_id_distribution.parse_args([
'-i',
os.path.dirname(os.path.abspath(__file__)) +
'/../test/test_user_ids.json', '-o',
os.path.dirname(os.path.abspath(__file__)) + '/../test/output.csv'
])
query_user_id_distribution.query_distribution(args.output, args.input)
with open(
os.path.dirname(os.path.abspath(__file__)) +
'/../test/test_user_ids.json') as f:
original_total_count = sum(1 for _ in f)
with open(
os.path.dirname(os.path.abspath(__file__)) +
'/../test/output.csv') as f:
reader = csv.DictReader(f)
for row in reader:
if row['id'] == 'all_users_total':
new_total_count = int(row['smapp_tweet_count'])
self.assertEqual(original_total_count, new_total_count)
self.tearDown()
def test_merge_json_parse_args(self):
args = query_user_id_distribution.parse_args(['-i', '~/data1.json', '-o', '~/data3.json'])
self.assertEqual(args.input,'~/data1.json')
self.assertEqual(args.output,'~/data3.json')
def test_merge_json_parse_args(self):
args = query_user_id_distribution.parse_args(
['-i', '~/data1.json', '-o', '~/data3.json'])
self.assertEqual(args.input, '~/data1.json')
self.assertEqual(args.output, '~/data3.json')