Sets the default ion_type when an IonPy* is constructed (#173)

amazon/ion/simple_types.py

@@ -140,10 +140,11 @@ def to_event(self, event_type, field_name=None, in_struct=False, depth=None):
                                   annotations=self.ion_annotations, depth=depth)
 
 
-def _ion_type_for(name, base_cls):
+def _ion_type_for(name, base_cls, ion_type=None):
     class IonPyValueType(base_cls, _IonNature):
         def __init__(self, *args, **kwargs):
             super(IonPyValueType, self).__init__(*args, **kwargs)
+            self.ion_type = ion_type
 
     IonPyValueType.__name__ = name
     IonPyValueType.__qualname__ = name
@@ -157,15 +158,16 @@ def __init__(self, *args, **kwargs):
 
 
 IonPyBool = IonPyInt
-IonPyFloat = _ion_type_for('IonPyFloat', float)
-IonPyDecimal = _ion_type_for('IonPyDecimal', Decimal)
+IonPyFloat = _ion_type_for('IonPyFloat', float, IonType.FLOAT)
+IonPyDecimal = _ion_type_for('IonPyDecimal', Decimal, IonType.DECIMAL)
 IonPyText = _ion_type_for('IonPyText', six.text_type)
 IonPyBytes = _ion_type_for('IonPyBytes', six.binary_type)
 
 
 class IonPySymbol(SymbolToken, _IonNature):
     def __init__(self, *args, **kwargs):
         super(IonPySymbol, self).__init__(*args, **kwargs)
+        self.ion_type = IonType.SYMBOL
 
     @staticmethod
     def _to_constructor_args(st):
@@ -180,6 +182,7 @@ def _to_constructor_args(st):
 class IonPyTimestamp(Timestamp, _IonNature):
     def __init__(self, *args, **kwargs):
         super(IonPyTimestamp, self).__init__(*args, **kwargs)
+        self.ion_type = IonType.TIMESTAMP
 
     @staticmethod
     def _to_constructor_args(ts):
@@ -222,5 +225,4 @@ def is_null(value):
 
 
 IonPyList = _ion_type_for('IonPyList', list)
-IonPyDict = _ion_type_for('IonPyDict', Multimap)
-
+IonPyDict = _ion_type_for('IonPyDict', Multimap, IonType.STRUCT)