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)