fix: rename new_columns to with_colummns

Author: FelixAbrahamsson

datastream/dataset.py

@@ -330,7 +330,7 @@ def group_split(
         }
 
 
-    def new_columns(
+    def with_columns(
         self: Dataset[T], **kwargs: Callable[pd.Dataframe, pd.Series]
     ) -> Dataset[T]:
         '''
@@ -340,7 +340,7 @@ def new_columns(
 
         >>> (
         ...     Dataset.from_dataframe(pd.DataFrame(dict(number=[1, 2, 3])))
-        ...     .new_columns(twice=lambda df: df['number'] * 2)
+        ...     .with_columns(twice=lambda df: df['number'] * 2)
         ...     .map(lambda row: row['twice'])
         ... )[-1]
         6
@@ -576,15 +576,15 @@ def test_subset():
     assert dataset[0]['number'] == numbers[2]
 
 
-def test_new_columns():
+def test_with_columns():
     from pytest import raises
 
     with raises(ValueError):
         dataset = (
             Dataset.from_dataframe(pd.DataFrame(dict(
                 key=np.arange(100),
             )))
-            .new_columns(key=lambda df: df['key'] * 2)
+            .with_columns(key=lambda df: df['key'] * 2)
         )