Added placeholder tests for other FIBContext helper functions, added test for '_number_from_name', and fixed broken regex search

Author: tieneupin

src/murfey/client/contexts/fib.py

@@ -56,7 +56,7 @@ def _number_from_name(name: str) -> int:
     """
     return (
         int(match.group(1))
-        if (match := re.search(r"\(([\d+])\)", name)) is not None
+        if (match := re.search(r"^[\w\s]+\((\d+)\)$", name)) is not None
         else 1
     )
 

tests/client/contexts/test_fib.py

@@ -6,7 +6,7 @@
 import pytest
 from pytest_mock import MockerFixture
 
-from murfey.client.contexts.fib import FIBContext
+from murfey.client.contexts.fib import FIBContext, _number_from_name
 
 # -------------------------------------------------------------------------------------
 # FIBContext test utilty functions and fixtures
@@ -193,6 +193,34 @@ def fib_maps_images(tmp_path: Path):
 # -------------------------------------------------------------------------------------
 # Tests
 # -------------------------------------------------------------------------------------
+
+
+@pytest.mark.parametrize(
+    "test_params",
+    (  # File name | Expected number
+        # AutoTEM examples
+        ("Lamella", 1),
+        ("Lamella (2)", 2),
+        ("Lamella (12)", 12),
+        # Maps examples
+        ("Electron Snapshot", 1),
+        ("Electron Snapshot (3)", 3),
+        ("Electron Snapshot (21)", 21),
+    ),
+)
+def test_number_from_name(test_params: tuple[str, int]):
+    name, number = test_params
+    assert _number_from_name(name) == number
+
+
+def test_get_source():
+    pass
+
+
+def test_file_transferred_to():
+    pass
+
+
 def test_fib_autotem_context():
     pass