improves typings

Author: kaelzhang

stock_pandas/dataframe.py

@@ -340,7 +340,10 @@ def _get_or_calc_series(
         return name, array
 
     def _fulfill_series(self, column_name: str) -> ndarray:
-        column_info = self._stock_columns_info_map.get(column_name)
+        # Since `column_name` always exists logically,
+        #   we could safely get by dict[key]
+        column_info = self._stock_columns_info_map[column_name]
+
         size = len(self)
 
         array = self.get_column(column_name).to_numpy()

stock_pandas/meta/cumulator.py

@@ -92,8 +92,8 @@ def cum_append(
     name = other[0].name
     other = DataFrame(other)
 
-    if (df.columns.get_indexer(other.columns) >= 0).all():
-        other = other.reindex(columns=df.columns)
+    if (df.columns.get_indexer(other.columns) >= 0).all():  # type: ignore
+        other = other.reindex(columns=df.columns)  # type: ignore
 
     length = len(df[name:]) if len(df) else 0
     if length:
@@ -129,7 +129,7 @@ def update(
         self,
         df: 'MetaDataFrame',
         source,
-        source_cumulator: '_Cumulator' = None,
+        source_cumulator: Optional['_Cumulator'] = None,
         date_col: Optional[str] = None,
         to_datetime_kwargs: dict = {},
         time_frame: TimeFrameArg = None,

test/test_truncated.py

@@ -97,7 +97,7 @@ def test_invalid_slicing(stock: StockDataFrame):
         # However, it might never happen,
         # but there is a assertion in super()._slice
         # we have to test about this case
-        stock._slice({})
+        stock._slice({})  # type: ignore
 
     assert stock._stock_indexer_slice is None