diff --git a/rbc/common.thrift b/rbc/common.thrift index 62ba5dad..78379403 100644 --- a/rbc/common.thrift +++ b/rbc/common.thrift @@ -27,7 +27,8 @@ enum TDatumType { TINYINT, GEOMETRY, GEOGRAPHY, - MULTILINESTRING + MULTILINESTRING, + MULTIPOINT } enum TEncodingType { diff --git a/rbc/heavydb/remoteheavydb.py b/rbc/heavydb/remoteheavydb.py index 30dfa705..426594b9 100644 --- a/rbc/heavydb/remoteheavydb.py +++ b/rbc/heavydb/remoteheavydb.py @@ -899,6 +899,7 @@ def _get_ext_arguments_map(self): 'GeoMultiPolygon': typemap['TExtArgumentType'].get( 'GeoMultiPolygon'), 'GeoMultiLineString': typemap['TExtArgumentType'].get('GeoMultiLineString'), + 'GeoMultiPoint': typemap['TExtArgumentType'].get('GeoMultiPoint'), 'TextEncodingNone': typemap['TExtArgumentType'].get('TextEncodingNone'), 'TextEncodingDict': typemap['TExtArgumentType'].get('TextEncodingDict'), 'Timestamp': typemap['TExtArgumentType'].get('Timestamp'), @@ -922,8 +923,17 @@ def _get_ext_arguments_map(self): ('double', 'Double'), ('TextEncodingDict', 'TextEncodingDict'), ('TextEncodingNone', 'TextEncodingNone'), - ('Timestamp', 'Timestamp')]: + ('Timestamp', 'Timestamp'), + ('GeoLineString', 'GeoLineString'), + ('GeoPolygon', 'GeoPolygon'), + ('GeoMultiPoint', 'GeoMultiPoint'), + ('GeoMultiLineString', 'GeoMultiLineString'), + ('GeoPoint', 'GeoPoint'), + ('GeoMultiPolygon', 'GeoMultiPolygon'), + ]: ext_arguments_map[f'Column<{T}>'] = typemap['TExtArgumentType'].get(f'Column{Tname}') + if T == 'Timestamp': + continue ext_arguments_map[f'ColumnList<{T}>'] = typemap['TExtArgumentType'].get( f'ColumnList{Tname}') ext_arguments_map[f'ColumnArray<{T}>'] = typemap['TExtArgumentType'].get( @@ -944,12 +954,15 @@ def _get_ext_arguments_map(self): ('TextEncodingDict', 'TextEncodingDict'), ('Timestamp', 'Timestamp'), ]: - ext_arguments_map['HeavyDBArrayType<%s>' % ptr_type] \ - = ext_arguments_map.get('Array<%s>' % T) ext_arguments_map['HeavyDBColumnType<%s>' % ptr_type] \ = ext_arguments_map.get('Column<%s>' % T) ext_arguments_map['HeavyDBOutputColumnType<%s>' % ptr_type] \ = ext_arguments_map.get('Column<%s>' % T) + if T == 'Timestamp': + # timestamp is only defined for Columns + continue + ext_arguments_map['HeavyDBArrayType<%s>' % ptr_type] \ + = ext_arguments_map.get('Array<%s>' % T) ext_arguments_map['HeavyDBColumnListType<%s>' % ptr_type] \ = ext_arguments_map.get('ColumnList<%s>' % T) ext_arguments_map['HeavyDBOutputColumnListType<%s>' % ptr_type] \ diff --git a/rbc/tests/__init__.py b/rbc/tests/__init__.py index bdf9615f..2a6d9279 100644 --- a/rbc/tests/__init__.py +++ b/rbc/tests/__init__.py @@ -308,7 +308,7 @@ def require_version(version, message=None, label=None): if not available_version: pytest.skip(reason) # Requires update when heavydb-internal bumps up version number: - current_development_version = Version("6.3.0") + current_development_version = Version("6.4.0") curr_version = Version('.'.join(map(str, available_version[:2])))