Add tests.

Author: JoeZiminski

tests/tests_integration/test_configs.py

@@ -48,10 +48,12 @@ def test_warning_on_startup(self, no_cfg_project):
             "Use make_config_file() to setup before continuing."
         )
 
-    def test_empty_project_name(self):
+    def test_empty_project_name(self, tmp_path):
         """
         Empty project names ("") are not allowed.
         """
+        os.chdir(tmp_path)  # avoids messy backup folder creation
+
         with pytest.raises(NeuroBlueprintError) as e:
             DataShuttle("")
 

tests/tests_tui/test_tui_configs.py

@@ -360,6 +360,44 @@ async def test_bad_configs_screen_input(self, empty_project_paths):
             )
             await pilot.pause()
 
+    # -------------------------------------------------------------------------
+    # Test project name is number
+    # -------------------------------------------------------------------------
+
+    @pytest.mark.asyncio
+    async def test_project_name_is_number(self):
+        """
+        Make a project that has a number name, and check the project screen
+        can be loaded.
+        """
+        app = TuiApp()
+        async with app.run_test(size=self.tui_size()) as pilot:
+
+            # Set up a project with a numerical project name
+            project_name = "123"
+
+            await self.scroll_to_click_pause(
+                pilot, "#mainwindow_new_project_button"
+            )
+
+            await self.fill_input(pilot, "#configs_name_input", project_name)
+            await self.fill_input(pilot, "#configs_local_path_input", "a")
+            await self.fill_input(pilot, "#configs_central_path_input", "b")
+            await self.scroll_to_click_pause(
+                pilot, "#configs_save_configs_button"
+            )
+
+            # Go back to main menu and load the project screen
+            await self.close_messagebox(pilot)
+
+            await self.scroll_to_click_pause(pilot, "#all_main_menu_buttons")
+
+            await self.check_and_click_onto_existing_project(
+                pilot, project_name
+            )
+
+            await pilot.pause()
+
     # -------------------------------------------------------------------------
     # Helpers
     # -------------------------------------------------------------------------

tests/tests_tui/tui_base.py

@@ -166,7 +166,7 @@ async def check_and_click_onto_existing_project(self, pilot, project_name):
         assert len(pilot.app.screen.project_names) == 1
         assert project_name in pilot.app.screen.project_names
 
-        await pilot.click(f"#{project_name}")
+        await pilot.click(f"#safety_prefix_{project_name}")
         await pilot.pause()
 
         assert isinstance(pilot.app.screen, ProjectManagerScreen)