add simple benchmark

brainglobe · May 31, 2024 · 06e85d2 · 06e85d2
1 parent 86c292c
commit 06e85d2
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 0 deletions.
diff --git a/benchmarks/benchmarks/brainglobe_atlasapi.py b/benchmarks/benchmarks/brainglobe_atlasapi.py
@@ -0,0 +1,19 @@
+from brainglobe_workflows.brainglobe_atlasapi.create_mouse_atlas import (
+    create_mouse_atlas,
+)
+
+
+class TimeBenchmark:
+    # Timing attributes
+    timeout = 3600  # default: 60 s
+    version = (
+        None  # benchmark version. Default:None (i.e. hash of source code)
+    )
+    warmup_time = 0.1  # seconds
+    rounds = 2
+    repeat = 0
+    sample_time = 0.01  # default: 10 ms = 0.01 s;
+    min_run_count = 2  # default:2
+
+    def time_create_mouse_atlas(self):
+        create_mouse_atlas()
diff --git a/brainglobe_workflows/brainglobe_atlasapi/__init__.py b/brainglobe_workflows/brainglobe_atlasapi/__init__.py
diff --git a/brainglobe_workflows/brainglobe_atlasapi/create_mouse_atlas.py b/brainglobe_workflows/brainglobe_atlasapi/create_mouse_atlas.py
@@ -0,0 +1,10 @@
+from brainglobe_atlasapi import BrainGlobeAtlas
+
+
+def create_mouse_atlas():
+    atlas = BrainGlobeAtlas("allen_mouse_100um")
+    print(atlas.name)
+
+
+if __name__ == "__main__":
+    create_mouse_atlas()