added a key / ref unit test and updated unit test script to optionally skip clean up

Author: dschwartz-ftadvisory

test/python_unit_tests/model/test_date_serialization.py

@@ -8,7 +8,7 @@
 
 def test_date_serialization():
     '''test date serialization'''
-    dt_in = DateSerializatonTest(dateValue=DateTimeWithMeta(datetime.datetime(2025, 10, 30)))
+    dt_in = DateSerializatonTest(dateValue=datetime.datetime(2025, 10, 30))
     dt_in.validate_model()
     dt_str = dt_in.rune_serialize()
     print('dt_str:', dt_str)

test/python_unit_tests/model/test_key_ref.py

@@ -0,0 +1,14 @@
+'''key ref unit test'''
+from rune.runtime.metadata import IntWithMeta, Reference
+
+from rosetta_dsl.test.model.key_ref.A import A
+from rosetta_dsl.test.model.key_ref.B import B
+
+def test_key_ref():
+    '''test key ref'''
+    a = A(aValue=IntWithMeta(value=1, key="key-123"))
+    b = B(aReference=Reference(target=a, ext_key="key-123"))
+    assert(len(b.validate_model())== 0)
+
+if __name__ == "__main__":
+    test_key_ref()

test/python_unit_tests/rosetta/KeyRefTest.rosetta

@@ -0,0 +1,9 @@
+namespace rosetta_dsl.test.model.key_ref : <"generate Python unit tests from Rosetta.">
+
+type A:
+    [metadata key]
+    aValue int (1..1)
+
+type B:
+    aReference A (1..1) 
+        [metadata reference]

test/python_unit_tests/run_python_unit_tests.sh

@@ -1,27 +1,64 @@
 #!/bin/bash
-function error
-{
+
+function usage {
+  cat <<EOF
+Usage: $(basename "$0") [options]
+
+Options:
+  -k, --no-clean, --skip-clean, --keep-venv  Skip the cleanup step (leave venv active; do not run cleanup script)
+  -h, --help                                  Show this help
+Env:
+  SKIP_CLEANUP=1                              Same as --no-clean
+EOF
+}
+
+function error {
     echo
     echo "***************************************************************************"
     echo "*                                                                         *"
     echo "*                     DEV ENV Initialization FAILED!                      *"
     echo "*                                                                         *"
     echo "***************************************************************************"
     echo
-    exit -1
+    exit 1
 }
 
+# Default: perform cleanup unless asked not to
+CLEANUP=1
+# Env toggle
+if [[ "${SKIP_CLEANUP:-}" == "1" || "${SKIP_CLEANUP:-}" == "true" ]]; then
+  CLEANUP=0
+fi
+# CLI options
+while [[ $# -gt 0 ]]; do
+  case "$1" in
+    -k|--no-clean|--skip-clean|--keep-venv)
+      CLEANUP=0
+      shift
+      ;;
+    -h|--help)
+      usage
+      exit 0
+      ;;
+    *)
+      echo "Unknown option: $1"
+      usage
+      exit 2
+      ;;
+  esac
+done
+
 export PYTHONDONTWRITEBYTECODE=1
 
-type -P python > /dev/null && PYEXE=python || PYEXE=python3
-if ! $PYEXE -c 'import sys; assert sys.version_info >= (3,11)' > /dev/null 2>&1; then
-        echo "Found $($PYEXE -V)"
-        echo "Expecting at least python 3.11 - exiting!"
-        exit 1
+type -P python >/dev/null && PYEXE=python || PYEXE=python3
+if ! $PYEXE -c 'import sys; assert sys.version_info >= (3,11)' >/dev/null 2>&1; then
+  echo "Found $($PYEXE -V)"
+  echo "Expecting at least python 3.11 - exiting!"
+  exit 1
 fi
 
 MY_PATH="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
-cd ${MY_PATH} || error
+cd "${MY_PATH}" || error
 PROJECT_ROOT_PATH="$MY_PATH/../.."
 PYTHON_SETUP_PATH="$MY_PATH/../python_setup"
 
@@ -48,27 +85,34 @@ java -cp "$JAR_PATH" com.regnosys.rosetta.generator.python.PythonCodeGeneratorCL
   -t "$PYTHON_TESTS_TARGET_PATH"
 JAVA_EXIT_CODE=$?
 if [[ $JAVA_EXIT_CODE -ne 0 ]]; then
-    echo "Java program returned exit code $JAVA_EXIT_CODE. Stopping script."
-    exit 1
+  echo "Java program returned exit code $JAVA_EXIT_CODE. Stopping script."
+  exit 1
 fi
 
 echo "***** setting up common environment"
+# shellcheck disable=SC1090
 source "$PYTHON_SETUP_PATH/setup_python_env.sh"
 
 echo "***** activating virtual environment"
 VENV_NAME=".pyenv"
+# shellcheck disable=SC1090
 source "$PROJECT_ROOT_PATH/$VENV_NAME/${PY_SCRIPTS}/activate" || error
 
 # package and install generated Python
-cd $PYTHON_TESTS_TARGET_PATH
-$PYEXE -m pip wheel --no-deps --only-binary :all: . || processError
+cd "$PYTHON_TESTS_TARGET_PATH" || error
+$PYEXE -m pip wheel --no-deps --only-binary :all: . || error
 $PYEXE -m pip install python_rosetta_dsl-0.0.0-py3-none-any.whl
 
 # run tests
 echo "***** run unit tests"
-cd "$MY_PATH"
+cd "$MY_PATH" || error
 $PYEXE -m pytest -p no:cacheprovider "$MY_PATH"
 
-echo "***** cleanup"
-deactivate
-source "$PYTHON_SETUP_PATH/cleanup_python_env.sh"
+if (( CLEANUP )); then
+  echo "***** cleanup"
+  deactivate 2>/dev/null || true
+  # shellcheck disable=SC1090
+  source "$PYTHON_SETUP_PATH/cleanup_python_env.sh"
+else
+  echo "***** skipping cleanup (requested)"
+fi