def test_csv_to_json_list(self): self.setUp() args = csv_to_json.parse_args([ '-i', os.path.dirname(os.path.abspath(__file__)) + '/../test/test.csv', os.path.dirname(os.path.abspath(__file__)) + '/../test/test.csv', '-o', os.path.dirname(os.path.abspath(__file__)) + '/../test/output.json', '-f', 'consumer_key', 'consumer_secret', 'access_token', 'access_token_secret', '--jsonlist' ]) csv_to_json.csv_to_json_list(args.output, args.inputs, args.fieldnames, args.skipheader) #gotta compare line counts and not sizes. with open( os.path.dirname(os.path.abspath(__file__)) + '/../test/test.csv') as f: original_count = sum(1 for _ in f) with open( os.path.dirname(os.path.abspath(__file__)) + '/../test/output.json') as f: json_list = json.load(f) new_count = len(json_list) #ouput should be the same number of lines as one input since the merge merges the same file with same object ids 2x. self.assertEqual(2 * original_count, new_count) #delete output.json when test is done self.tearDown()
def test_merge_json_parse_args(self): args = csv_to_json.parse_args([ '-i', '~/data1.csv', '~/data2.csv', '-o', '~/data3.json', '-f', 'consumer_key', 'consumer_secret', 'access_token', 'access_token_secret' ]) self.assertEqual(set(args.inputs), set(['~/data1.csv', '~/data2.csv'])) self.assertEqual(args.output, '~/data3.json') self.assertEqual(args.fieldnames, [ 'consumer_key', 'consumer_secret', 'access_token', 'access_token_secret' ])
def test_csv_to_json(self): self.setUp() args = csv_to_json.parse_args(['-i', os.path.dirname(os.path.abspath(__file__))+'/../test/test.csv', os.path.dirname(os.path.abspath(__file__))+'/../test/test.csv', '-o', os.path.dirname(os.path.abspath(__file__))+'/../test/output.json', '-f', 'consumer_key', 'consumer_secret', 'access_token', 'access_token_secret']) csv_to_json.csv_to_json(args.output, args.inputs, args.fieldnames, args.skipheader) #gotta compare line counts and not sizes. with open(os.path.dirname(os.path.abspath(__file__))+'/../test/test.csv') as f: original_count = sum(1 for _ in f) with open(os.path.dirname(os.path.abspath(__file__))+'/../test/output.json') as f: new_count = sum(1 for _ in f) #ouput should be the same number of lines as one input since the merge merges the same file with same object ids 2x. self.assertEqual(2*original_count, new_count) #delete output.json when test is done self.tearDown()
def test_merge_json_parse_args(self): args = csv_to_json.parse_args(['-i', '~/data1.csv', '~/data2.csv', '-o', '~/data3.json', '-f', 'consumer_key', 'consumer_secret', 'access_token', 'access_token_secret']) self.assertEqual(set(args.inputs),set(['~/data1.csv', '~/data2.csv'])) self.assertEqual(args.output,'~/data3.json') self.assertEqual(args.fieldnames, ['consumer_key', 'consumer_secret', 'access_token', 'access_token_secret'])