Ensure function name is saved to JSON

Author: StephenNneji

ratapi/project.py

@@ -956,12 +956,16 @@ def make_data_dict(item):
             elif field == "custom_files":
 
                 def make_custom_file_dict(item):
-                    return {
+                    file_dict = {
                         "name": item.name,
                         "filename": item.filename,
                         "language": item.language,
                         "path": try_relative_to(item.path, filepath.parent),
                     }
+                    if item.name != item.function_name:
+                        file_dict["function_name"] = item.function_name
+
+                    return file_dict
 
                 json_dict["custom_files"] = [make_custom_file_dict(file) for file in attr]
 

tests/conftest.py

@@ -8537,3 +8537,11 @@ def absorption():
     """The project from the absorption example."""
     project, _ = ratapi.examples.absorption()
     return project
+
+
+@pytest.fixture
+def absorption_different_function():
+    """The project from the absorption example with a function name different from filename."""
+    project, _ = ratapi.examples.absorption()
+    project.custom_files[0].function_name = "test_func"
+    return project

tests/test_project.py

@@ -1616,6 +1616,7 @@ def test_wrap_extend(test_project, class_list: str, model_type: str, field: str,
         "domains_custom_layers",
         "domains_custom_xy",
         "absorption",
+        "absorption_different_function",
     ],
 )
 def test_save_load(project, request):