Add more benchmarks

• Regridding adapters

benchmarks/adapters/test_regrid.py

+import datetime as dt
+import unittest
+
+import pytest
+
+import finam as fm
+
+
+class TestRegrid(unittest.TestCase):
+    @pytest.fixture(autouse=True)
+    def setupBenchmark(self, benchmark):
+        self.benchmark = benchmark
+
+    def setup_adapter(self, grid1, grid2, adapter):
+        time = dt.datetime(2000, 1, 1)
+        self.data = fm.data.full(1.0, "test", fm.Info(time=time, grid=grid1), time=time)
+
+        self.source = fm.modules.CallbackGenerator(
+            callbacks={"Step": (lambda t: self.data, fm.Info(None, grid=grid1))},
+            start=time,
+            step=dt.timedelta(1.0),
+        )
+        self.adapter = adapter
+        self.source.initialize()
+        self.source.outputs["Step"] >> self.adapter
+
+        self.adapter.get_info(fm.Info(None, grid=grid2))
+        self.source.connect()
+        self.source.connect()
+        self.source.validate()
+
+    @pytest.mark.benchmark(group="adapters-regrid")
+    def test_regrid_nearest_01_32x16(self):
+        grid1 = fm.UniformGrid((32, 16))
+        grid2 = fm.UniformGrid((32, 16), origin=(0.25, 0.25))
+
+        self.setup_adapter(grid1, grid2, fm.adapters.RegridNearest())
+        _result = self.benchmark(
+            self.adapter.get_data, time=dt.datetime(2000, 1, 1), target=None
+        )
+
+    @pytest.mark.benchmark(group="adapters-regrid")
+    def test_regrid_nearest_02_512x256(self):
+        grid1 = fm.UniformGrid((512, 256))
+        grid2 = fm.UniformGrid((512, 256), origin=(0.25, 0.25))
+
+        self.setup_adapter(grid1, grid2, fm.adapters.RegridNearest())
+        _result = self.benchmark(
+            self.adapter.get_data, time=dt.datetime(2000, 1, 1), target=None
+        )
+
+    @pytest.mark.benchmark(group="adapters-regrid")
+    def test_regrid_nearest_03_2048x1024(self):
+        grid1 = fm.UniformGrid((2048, 1024))
+        grid2 = fm.UniformGrid((2048, 1024), origin=(0.25, 0.25))
+
+        self.setup_adapter(grid1, grid2, fm.adapters.RegridNearest())
+        _result = self.benchmark(
+            self.adapter.get_data, time=dt.datetime(2000, 1, 1), target=None
+        )
+
+    @pytest.mark.benchmark(group="adapters-regrid")
+    def test_regrid_linear_01_32x16(self):
+        grid1 = fm.UniformGrid((32, 16))
+        grid2 = fm.UniformGrid((32, 16), origin=(0.25, 0.25))
+
+        self.setup_adapter(grid1, grid2, fm.adapters.RegridLinear())
+        _result = self.benchmark(
+            self.adapter.get_data, time=dt.datetime(2000, 1, 1), target=None
+        )
+
+    @pytest.mark.benchmark(group="adapters-regrid")
+    def test_regrid_linear_02_512x256(self):
+        grid1 = fm.UniformGrid((512, 256))
+        grid2 = fm.UniformGrid((512, 256), origin=(0.25, 0.25))
+
+        self.setup_adapter(grid1, grid2, fm.adapters.RegridLinear())
+        _result = self.benchmark(
+            self.adapter.get_data, time=dt.datetime(2000, 1, 1), target=None
+        )
+
+    @pytest.mark.benchmark(group="adapters-regrid")
+    def test_regrid_linear_03_2048x1024(self):
+        grid1 = fm.UniformGrid((2048, 1024))
+        grid2 = fm.UniformGrid((2048, 1024), origin=(0.25, 0.25))
+
+        self.setup_adapter(grid1, grid2, fm.adapters.RegridLinear())
+        _result = self.benchmark(
+            self.adapter.get_data, time=dt.datetime(2000, 1, 1), target=None
+        )