def test_bad_csv_to_json(self):
'''
Test method to confirm behavior when bad CSV file is passed
'''
return_data, error_count = import_csv_to_json('fake_file')
self.assertEqual(return_data, [])
self.assertEqual(error_count, 1)
def test_csv_to_json(self):
'''
Test method to convert CSV files to JSON format
'''
directory = 'csv_data'
product_file = 'products.csv'
customer_file = 'customers.csv'
rentals_file = 'rentals.csv'
product_json_actual, _ = import_csv_to_json(
os.path.join(directory, product_file))
customer_json_actual, _ = import_csv_to_json(
os.path.join(directory, customer_file))
rentals_json_actual, _ = import_csv_to_json(
os.path.join(directory, rentals_file))
self.assertEqual(self.product_json, product_json_actual)
self.assertEqual(self.customer_json, customer_json_actual)
self.assertEqual(self.rentals_json, rentals_json_actual)