def test_converts_given_columns(self):
spec = Mock()
spec.start_line = 0
spec.header_lines = []
fstor = Mock(spec=DelayedOpen)
source = ShapefileSource(spec, fstor)
column1 = ('name1', 'int:10')
column2 = ('name2', 'str:10')
converted_column1 = {'name': 'name1', 'type': 'int'}
converted_column2 = {'name': 'name2', 'type': 'str'}
shapefile_columns = OrderedDict([column1, column2])
ret = source._get_columns(shapefile_columns)
self.assertIn(converted_column1, ret)
self.assertIn(converted_column2, ret)
def test_converts_given_columns(self):
spec = Mock()
spec.start_line = 0
spec.header_lines = []
fstor = Mock(spec=DelayedOpen)
source = ShapefileSource(spec, fstor)
column1 = ('name1', 'int:10')
column2 = ('name2', 'str:10')
converted_column1 = {'name': 'name1', 'type': 'int'}
converted_column2 = {'name': 'name2', 'type': 'str'}
shapefile_columns = OrderedDict([column1, column2])
ret = source._get_columns(shapefile_columns)
self.assertIn(converted_column1, ret)
self.assertIn(converted_column2, ret)
def test_extends_with_id_and_geometry(self):
spec = Mock()
spec.start_line = 0
spec.header_lines = []
fstor = Mock(spec=DelayedOpen)
source = ShapefileSource(spec, fstor)
shapefile_columns = OrderedDict()
ret = source._get_columns(shapefile_columns)
self.assertEqual(len(ret), 2)
names = [x['name'] for x in ret]
self.assertIn('id', names)
self.assertIn('geometry', names)
types = [x['type'] for x in ret]
self.assertIn('geometry_type', types)
def test_extends_with_id_and_geometry(self):
spec = Mock()
spec.start_line = 0
spec.header_lines = []
fstor = Mock(spec=DelayedOpen)
source = ShapefileSource(spec, fstor)
shapefile_columns = OrderedDict()
ret = source._get_columns(shapefile_columns)
self.assertEqual(len(ret), 2)
names = [x['name'] for x in ret]
self.assertIn('id', names)
self.assertIn('geometry', names)
types = [x['type'] for x in ret]
self.assertIn('geometry_type', types)
