Set up benchmarks

benchmarks/data/test_tools.py

@@ -8,6 +8,8 @@ from finam.data import (
     assign_time,
     check,
     full,
+    full_like,
+    get_magnitude,
     get_time,
     get_units,
     has_time,
@@ -23,17 +25,17 @@ class TestCheckXarray(unittest.TestCase):
     def setupBenchmark(self, benchmark):
         self.benchmark = benchmark
 
-    @pytest.mark.benchmark(group="tools")
+    @pytest.mark.benchmark(group="data-tools")
     def test_check_xarray_01_2x1(self):
         time = dt.datetime(2000, 1, 1)
         info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
         xdata = full(0.0, "test", info, time)
         _result = self.benchmark(check, xdata=xdata, name="test", info=info, time=time)
 
-    @pytest.mark.benchmark(group="tools")
-    def test_check_xarray_02_128x64(self):
+    @pytest.mark.benchmark(group="data-tools")
+    def test_check_xarray_02_512x256(self):
         time = dt.datetime(2000, 1, 1)
-        info = fm.Info(time=time, grid=fm.UniformGrid((128, 64)), units="m")
+        info = fm.Info(time=time, grid=fm.UniformGrid((512, 256)), units="m")
         xdata = full(0.0, "test", info, time)
         _result = self.benchmark(check, xdata=xdata, name="test", info=info, time=time)
 
@@ -43,7 +45,7 @@ class TestToXarray(unittest.TestCase):
     def setupBenchmark(self, benchmark):
         self.benchmark = benchmark
 
-    @pytest.mark.benchmark(group="tools")
+    @pytest.mark.benchmark(group="data-tools")
     def test_to_xarray_01_2x1(self):
         time = dt.datetime(2000, 1, 1)
         info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
@@ -53,17 +55,17 @@ class TestToXarray(unittest.TestCase):
             to_xarray, data=data, name="test", info=info, time=time
         )
 
-    @pytest.mark.benchmark(group="tools")
-    def test_to_xarray_02_128x64(self):
+    @pytest.mark.benchmark(group="data-tools")
+    def test_to_xarray_02_512x256(self):
         time = dt.datetime(2000, 1, 1)
-        info = fm.Info(time=time, grid=fm.UniformGrid((128, 64)), units="m")
+        info = fm.Info(time=time, grid=fm.UniformGrid((512, 256)), units="m")
         xdata = full(0.0, "test", info, time)
         data = strip_data(xdata)
         _result = self.benchmark(
             to_xarray, data=data, name="test", info=info, time=time
         )
 
-    @pytest.mark.benchmark(group="tools")
+    @pytest.mark.benchmark(group="data-tools")
     def test_to_xarray_03_2048x1024(self):
         time = dt.datetime(2000, 1, 1)
         info = fm.Info(time=time, grid=fm.UniformGrid((2048, 1024)), units="m")
@@ -74,26 +76,77 @@ class TestToXarray(unittest.TestCase):
         )
 
 
+class TestFull(unittest.TestCase):
+    @pytest.fixture(autouse=True)
+    def setupBenchmark(self, benchmark):
+        self.benchmark = benchmark
+
+    @pytest.mark.benchmark(group="data-tools")
+    def test_full_01_2x1(self):
+        time = dt.datetime(2000, 1, 1)
+        info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
+        _result = self.benchmark(full, value=0.0, name="test", info=info, time=time)
+
+    @pytest.mark.benchmark(group="data-tools")
+    def test_full_02_512x256(self):
+        time = dt.datetime(2000, 1, 1)
+        info = fm.Info(time=time, grid=fm.UniformGrid((512, 256)), units="m")
+        _result = self.benchmark(full, value=0.0, name="test", info=info, time=time)
+
+    @pytest.mark.benchmark(group="data-tools")
+    def test_full_03_2048x1024(self):
+        time = dt.datetime(2000, 1, 1)
+        info = fm.Info(time=time, grid=fm.UniformGrid((2048, 1024)), units="m")
+        _result = self.benchmark(full, value=0.0, name="test", info=info, time=time)
+
+
+class TestFullLike(unittest.TestCase):
+    @pytest.fixture(autouse=True)
+    def setupBenchmark(self, benchmark):
+        self.benchmark = benchmark
+
+    @pytest.mark.benchmark(group="data-tools")
+    def test_full_like_01_2x1(self):
+        time = dt.datetime(2000, 1, 1)
+        info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
+        xdata = full(0.0, "test", info, time)
+        _result = self.benchmark(full_like, xdata=xdata, value=0.0)
+
+    @pytest.mark.benchmark(group="data-tools")
+    def test_full_like_02_512x256(self):
+        time = dt.datetime(2000, 1, 1)
+        info = fm.Info(time=time, grid=fm.UniformGrid((512, 256)), units="m")
+        xdata = full(0.0, "test", info, time)
+        _result = self.benchmark(full_like, xdata=xdata, value=0.0)
+
+    @pytest.mark.benchmark(group="data-tools")
+    def test_full_like_03_2048x1024(self):
+        time = dt.datetime(2000, 1, 1)
+        info = fm.Info(time=time, grid=fm.UniformGrid((2048, 1024)), units="m")
+        xdata = full(0.0, "test", info, time)
+        _result = self.benchmark(full_like, xdata=xdata, value=0.0)
+
+
 class TestTimeTools(unittest.TestCase):
     @pytest.fixture(autouse=True)
     def setupBenchmark(self, benchmark):
         self.benchmark = benchmark
 
-    @pytest.mark.benchmark(group="tools")
+    @pytest.mark.benchmark(group="data-tools")
     def test_strip_time(self):
         time = dt.datetime(2000, 1, 1)
         info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
         xdata = full(0.0, "test", info, time)
         _result = self.benchmark(strip_time, xdata=xdata)
 
-    @pytest.mark.benchmark(group="tools")
+    @pytest.mark.benchmark(group="data-tools")
     def test_assign_time_update(self):
         time = dt.datetime(2000, 1, 1)
         info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
         xdata = full(0.0, "test", info, time)
         _result = self.benchmark(assign_time, xdata=xdata, time=dt.datetime(2000, 1, 2))
 
-    @pytest.mark.benchmark(group="tools")
+    @pytest.mark.benchmark(group="data-tools")
     def test_assign_time_add(self):
         time = dt.datetime(2000, 1, 1)
         info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
@@ -101,14 +154,14 @@ class TestTimeTools(unittest.TestCase):
         xdata = strip_time(xdata)
         _result = self.benchmark(assign_time, xdata=xdata, time=dt.datetime(2000, 1, 2))
 
-    @pytest.mark.benchmark(group="tools")
+    @pytest.mark.benchmark(group="data-tools")
     def test_get_time(self):
         time = dt.datetime(2000, 1, 1)
         info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
         xdata = full(0.0, "test", info, time)
         _result = self.benchmark(get_time, xdata=xdata)
 
-    @pytest.mark.benchmark(group="tools")
+    @pytest.mark.benchmark(group="data-tools")
     def test_has_time(self):
         time = dt.datetime(2000, 1, 1)
         info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
@@ -121,30 +174,51 @@ class TestUnitsTools(unittest.TestCase):
     def setupBenchmark(self, benchmark):
         self.benchmark = benchmark
 
-    @pytest.mark.benchmark(group="tools")
+    @pytest.mark.benchmark(group="data-tools")
     def test_get_units(self):
         time = dt.datetime(2000, 1, 1)
         info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
         xdata = full(0.0, "test", info, time)
         _result = self.benchmark(get_units, xdata=xdata)
 
-    @pytest.mark.benchmark(group="tools")
-    def test_to_units_2x1(self):
+    @pytest.mark.benchmark(group="data-tools")
+    def test_to_units_01_2x1(self):
         time = dt.datetime(2000, 1, 1)
         info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
         xdata = full(0.0, "test", info, time)
         _result = self.benchmark(to_units, xdata=xdata, units="in")
 
-    @pytest.mark.benchmark(group="tools")
-    def test_to_units_128x64(self):
+    @pytest.mark.benchmark(group="data-tools")
+    def test_to_units_02_512x256(self):
         time = dt.datetime(2000, 1, 1)
-        info = fm.Info(time=time, grid=fm.UniformGrid((128, 64)), units="m")
+        info = fm.Info(time=time, grid=fm.UniformGrid((512, 256)), units="m")
         xdata = full(0.0, "test", info, time)
         _result = self.benchmark(to_units, xdata=xdata, units="in")
 
-    @pytest.mark.benchmark(group="tools")
-    def test_to_units_2048x1024(self):
+    @pytest.mark.benchmark(group="data-tools")
+    def test_to_units_03_2048x1024(self):
         time = dt.datetime(2000, 1, 1)
         info = fm.Info(time=time, grid=fm.UniformGrid((2048, 1024)), units="m")
         xdata = full(0.0, "test", info, time)
         _result = self.benchmark(to_units, xdata=xdata, units="in")
+
+    @pytest.mark.benchmark(group="data-tools")
+    def test_get_magnitude_01_2x1(self):
+        time = dt.datetime(2000, 1, 1)
+        info = fm.Info(time=time, grid=fm.UniformGrid((2, 1)), units="m")
+        xdata = full(0.0, "test", info, time)
+        _result = self.benchmark(get_magnitude, xdata=xdata)
+
+    @pytest.mark.benchmark(group="data-tools")
+    def test_get_magnitude_02_512x256(self):
+        time = dt.datetime(2000, 1, 1)
+        info = fm.Info(time=time, grid=fm.UniformGrid((512, 256)), units="m")
+        xdata = full(0.0, "test", info, time)
+        _result = self.benchmark(get_magnitude, xdata=xdata)
+
+    @pytest.mark.benchmark(group="data-tools")
+    def test_get_magnitude_03_2048x1024(self):
+        time = dt.datetime(2000, 1, 1)
+        info = fm.Info(time=time, grid=fm.UniformGrid((2048, 1024)), units="m")
+        xdata = full(0.0, "test", info, time)
+        _result = self.benchmark(get_magnitude, xdata=xdata)