Some more fixes for column label types handling (#45)

* Some more fixes for column label types handling

* typo

Author: trangevi

inference_schema/parameter_types/pandas_parameter_type.py

@@ -27,7 +27,7 @@ def __init__(self, sample_input, enforce_column_type=True, enforce_shape=True, a
         :param enforce_shape: Enforce that input shape must match that of the provided sample when `deserialize_input`
             is called.
         :type enforce_shape: bool
-        :param apply_column_names: Apply column names fromt he provided sample onto the input when `deserialize_input`
+        :param apply_column_names: Apply column names from the provided sample onto the input when `deserialize_input`
             is called.
         :type apply_column_names: bool
         :param orient: The Pandas orient to use when converting between a json object and a DataFrame. Possible orients
@@ -67,13 +67,13 @@ def deserialize_input(self, input_data):
 
         data_frame = pd.read_json(json.dumps(input_data), orient=self.orient)
 
-        if self.apply_column_names and isinstance(input_data, list) and not isinstance(input_data[0], dict):
+        if self.apply_column_names:
             data_frame.columns = self.sample_input.columns.copy()
 
         if self.enforce_column_type:
             sample_input_column_types = self.sample_input.dtypes.to_dict()
             converted_types = {x: sample_input_column_types.get(x, object) for x in data_frame.columns}
-            data_frame = data_frame.astype(dtype=converted_types, copy=False)
+            data_frame = data_frame.astype(dtype=converted_types)
 
         if self.enforce_shape:
             expected_shape = self.sample_input.shape

tests/test_pandas_parameter_type.py

@@ -39,10 +39,11 @@ def test_pandas_timestamp_handling(self, decorated_pandas_datetime_func):
         result = decorated_pandas_datetime_func(**pandas_input)
         assert_frame_equal(result, datetime)
 
-    def test_pandas_multi_type_columns_labels_handling(self, decorated_pandas_func_multi_type_column_labels,
-                                                       pandas_sample_input_multi_type_column_labels):
-        result = decorated_pandas_func_multi_type_column_labels(pandas_sample_input_multi_type_column_labels)
-        assert_frame_equal(result, pandas_sample_input_multi_type_column_labels)
+    def test_pandas_multi_type_columns_labels_handling(self, decorated_pandas_func_multi_type_column_labels):
+        pandas_input = {'name': ['Sarah', 'John'], 1: ['WA', 'CA']}
+        result = decorated_pandas_func_multi_type_column_labels(pandas_input)
+        expected_result = pd.DataFrame(pandas_input)
+        assert_frame_equal(result, expected_result)
 
 
 class TestNestedType(object):