def import_csv(self, data_source=''): """ Imports objects from csv file. """ if data_source == '': # data_source = settings.RESOURCE_GRAPH_LOCATIONS print '*' * 80 print 'No data source indicated. Please rerun command with \'-s\' parameter.' print '*' * 80 if isinstance(data_source, basestring): data_source = [data_source] for path in data_source: if os.path.isfile(os.path.join(path)): CSVFileImporter(path).import_all() else: file_paths = [ file_path for file_path in os.listdir(path) if file_path.endswith('.csv') ] for file_path in file_paths: CSVFileImporter(os.path.join(path, file_path)).import_all()
def test_n_1(self): og_tile_count = TileModel.objects.count() CSVFileImporter('tests/fixtures/data/csv/cardinality_test_data/n-1.csv' ).import_all() new_tile_count = TileModel.objects.count() tile_difference = new_tile_count - og_tile_count self.assertEqual(tile_difference, 4)