Dataset insertion benchmark (#7223)

* Add a benchmark for dataset element insertion

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update asv_bench/benchmarks/dataset_creation.py

* Rework the benchmark

* lint before the bot

* Update asv_bench/benchmarks/dataset_creation.py

Co-authored-by: Deepak Cherian <[email protected]>

* Update asv_bench/benchmarks/dataset_creation.py

Co-authored-by: Deepak Cherian <[email protected]>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Lint

* Rename the benchmark

* Rename benchmark

* Update asv_bench/benchmarks/dataset_in_memory_operation.py

Co-authored-by: Maximilian Roos <[email protected]>

* Update and rename dataset_in_memory_operation.py to merge.py

* add back elements

* Only add a single variable

* Give the parameter a name

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Illviljan <[email protected]>
Co-authored-by: Deepak Cherian <[email protected]>
Co-authored-by: Maximilian Roos <[email protected]>

Author: 5 people

asv_bench/benchmarks/merge.py

@@ -0,0 +1,18 @@
+import xarray as xr
+
+
+class DatasetAddVariable:
+    param_names = ["existing_elements"]
+    params = [[0, 10, 100, 1000]]
+
+    def setup(self, existing_elements):
+        self.datasets = {}
+        # Dictionary insertion is fast(er) than xarray.Dataset insertion
+        d = {}
+        for i in range(existing_elements):
+            d[f"var{i}"] = i
+        self.dataset = xr.merge([d])
+
+    def time_variable_insertion(self, existin_elements):
+        dataset = self.dataset
+        dataset["new_var"] = 0