pythonfmu3 0.3.2__tar.gz → 0.3.3__tar.gz

{pythonfmu3-0.3.2/pythonfmu3.egg-info → pythonfmu3-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pythonfmu3
-Version: 0.3.2
+Version: 0.3.3
 Summary: A lightweight framework that enables the packaging of Python3.x code as co-simulation FMUs.
 Home-page: https://github.com/stephensmith25/PythonFMU3
 Author: Stephen Smith
@@ -173,6 +173,7 @@ To create a model exchange FMU you must inherit from `Fmi3SlaveBase` and one of
 ```python
 
 from pythonfmu3 import Fmi3Causality, Fmi3SlaveBase, Float64, ModelExchange
+from typing import List
 
 
 class PythonSlaveMX(Fmi3SlaveBase, ModelExchange):

{pythonfmu3-0.3.2 → pythonfmu3-0.3.3}/README.md

@@ -146,6 +146,7 @@ To create a model exchange FMU you must inherit from `Fmi3SlaveBase` and one of
 ```python
 
 from pythonfmu3 import Fmi3Causality, Fmi3SlaveBase, Float64, ModelExchange
+from typing import List
 
 
 class PythonSlaveMX(Fmi3SlaveBase, ModelExchange):

pythonfmu3-0.3.3/pythonfmu3/_version.py

@@ -0,0 +1 @@
+__version__ = "0.3.3"

{pythonfmu3-0.3.2 → pythonfmu3-0.3.3}/pythonfmu3/builder.py

@@ -13,7 +13,7 @@ from typing import Iterable, Optional, Tuple, Union
 from xml.dom.minidom import parseString
 from xml.etree.ElementTree import Element, SubElement, tostring
 from .osutil import get_lib_extension, get_platform
-from .fmi3slave import FMI3_MODEL_OPTIONS, Fmi3Slave, Fmi3SlaveBase
+from .fmi3slave import FMI3_MODEL_OPTIONS_COMMON, FMI3_MODEL_OPTIONS_COSIM, FMI3_MODEL_OPTIONS_MX, Fmi3Slave, Fmi3SlaveBase
 
 FilePath = Union[str, Path]
 HERE = Path(__file__).parent
@@ -137,16 +137,20 @@ class FmuBuilder:
             dest_file = dest / f"{model_identifier}.fmu"
 
             type_node = xml.find("CoSimulation")
-            option_names = [opt.name for opt in FMI3_MODEL_OPTIONS]
-            if type_node:
+            option_names = [opt.name for opt in FMI3_MODEL_OPTIONS_COMMON]
+            cosim_option_names = option_names.copy()
+            cosim_option_names.extend(opt.name for opt in FMI3_MODEL_OPTIONS_COSIM)
+            if type_node is not None:
                 for option, value in options.items():
-                    if option in option_names:
+                    if option in cosim_option_names:
                         type_node.set(option, str(value).lower())
 
             type_node = xml.find("ModelExchange")
-            if type_node:
+            mx_option_names = option_names.copy()
+            mx_option_names.extend(opt.name for opt in FMI3_MODEL_OPTIONS_MX)
+            if type_node is not None:
                 for option, value in options.items():
-                    if option in option_names:
+                    if option in mx_option_names:
                         type_node.set(option, str(value).lower())
 
             with zipfile.ZipFile(dest_file, "w") as zip_fmu:
@@ -247,7 +251,25 @@ def create_command_parser(parser: argparse.ArgumentParser):
         default=None
     )
 
-    for option in FMI3_MODEL_OPTIONS:
+    for option in FMI3_MODEL_OPTIONS_COMMON:
+        action = "store_false" if option.value else "store_true"
+        parser.add_argument(
+            f"--{option.cli}",
+            dest=option.name,
+            help=f"If given, {option.name}={action[6:]}",
+            action=action
+        )
+
+    for option in FMI3_MODEL_OPTIONS_COSIM:
+        action = "store_false" if option.value else "store_true"
+        parser.add_argument(
+            f"--{option.cli}",
+            dest=option.name,
+            help=f"If given, {option.name}={action[6:]}",
+            action=action
+        )
+    
+    for option in FMI3_MODEL_OPTIONS_MX:
         action = "store_false" if option.value else "store_true"
         parser.add_argument(
             f"--{option.cli}",

{pythonfmu3-0.3.2 → pythonfmu3-0.3.3}/pythonfmu3/fmi3slave.py

@@ -21,14 +21,22 @@ from .unit import Unit
 
 ModelOptions = namedtuple("ModelOptions", ["name", "value", "cli"])
 
-FMI3_MODEL_OPTIONS: List[ModelOptions] = [
+FMI3_MODEL_OPTIONS_COMMON: List[ModelOptions] = [
     ModelOptions("needsExecutionTool", True, "no-external-tool"),
-    ModelOptions("canHandleVariableCommunicationStepSize", True, "no-variable-step"),
     ModelOptions("canBeInstantiatedOnlyOncePerProcess", False, "only-one-per-process"),
     ModelOptions("canGetAndSetFMUState", False, "handle-state"),
     ModelOptions("canSerializeFMUState", False, "serialize-state")
 ]
 
+FMI3_MODEL_OPTIONS_COSIM: List[ModelOptions] = [
+    ModelOptions("canHandleVariableCommunicationStepSize", True, "no-variable-step"),
+]
+
+FMI3_MODEL_OPTIONS_MX: List[ModelOptions] = [
+    ModelOptions("needsCompletedIntegratorStep", False, "needs-completed-step"),
+]
+
+
 class Fmi3StepResult(NamedTuple):
     status: Fmi3Status = Fmi3Status.ok
     eventHandlingNeeded: bool = False
@@ -103,23 +111,26 @@ class Fmi3SlaveBase(object):
         root = Element("fmiModelDescription", attrib)
 
         options = dict()
-        for option in FMI3_MODEL_OPTIONS:
+        for option in FMI3_MODEL_OPTIONS_COMMON:
             value = model_options.get(option.name, option.value)
             options[option.name] = str(value).lower()
+        
         options["modelIdentifier"] = self.modelName
-        options["canNotUseMemoryManagementFunctions"] = "true"
+
+        options_cs = options.copy()
+        for option in FMI3_MODEL_OPTIONS_COSIM:
+            options_cs[option.name] = str(value).lower()
         
-        options_me = dict()
-        options_me["canGetAndSetFMUState"] = "true"
-        options_me["modelIdentifier"] = self.modelName
-        options_me["needsCompletedIntegratorStep"] = "false"
+        options_me = options.copy()
+        for option in FMI3_MODEL_OPTIONS_MX:
+            options_me[option.name] = str(value).lower()
 
         # check if we have cosim mixin or model exchange mixin
         if isinstance(self, ModelExchange):
             SubElement(root, "ModelExchange", attrib=options_me)
         
         if isinstance(self, CoSimulation):
-            SubElement(root, "CoSimulation", attrib=options)
+            SubElement(root, "CoSimulation", attrib=options_cs)
 
         if self.units:
             unit_defs = SubElement(root, "UnitDefinitions")

pythonfmu3-0.3.3/pythonfmu3/tests/test_docs.py

@@ -0,0 +1,98 @@
+import pytest
+import pathlib
+import re
+import tempfile
+
+def extract_python_code_from_md(md_file):
+    """Extract Python code blocks from markdown, excluding skip blocks"""
+    content = pathlib.Path(md_file).read_text()
+    
+    # Find all python code blocks that aren't marked as skip
+    python_blocks = []
+    
+    # Pattern to match python blocks but not python skip
+    pattern = r'```python(?!\s+skip)(?:[^\n]*)\n(.*?)\n```'
+    matches = re.findall(pattern, content, re.DOTALL)
+    
+    for match in matches:
+        # Skip blocks that are wrapped in skip-test comments
+        if '<!-- skip-test -->' not in content.split(match)[0][-100:]:
+            python_blocks.append(match)
+    
+    return python_blocks
+
+@pytest.mark.doc
+def test_readme():
+    test_dir = pathlib.Path(__file__).parent
+    fpath = test_dir.parent.parent / "README.md"
+    
+    # extract Python code
+    code_blocks = extract_python_code_from_md(fpath)
+
+    namespace = {}
+    exec("from pythonfmu3 import *", namespace)
+
+    for i, code in enumerate(code_blocks):
+        try:
+            exec(code, namespace)
+            print(f"✅ Code block {i+1} executed successfully")
+        except Exception as e:
+            pytest.fail(f"Code block {i+1} failed: {e}\nCode:\n{code}")
+    
+
+
+
+@pytest.mark.doc
+def test_cosimulation_examples():
+    """Test that all Python examples in usage.md actually work"""
+    test_dir = pathlib.Path(__file__).parent
+    usage_file = test_dir.parent.parent / "docs" / "usage.md"
+    
+    # Extract Python code
+    code_blocks = extract_python_code_from_md(usage_file)
+    
+    # Create a test namespace
+    namespace = {}
+    exec("from pythonfmu3 import *", namespace)
+    
+    # Execute each code block
+    for i, code in enumerate(code_blocks):
+        try:
+            exec(code, namespace)
+            print(f"✅ Code block {i+1} executed successfully")
+        except Exception as e:
+            pytest.fail(f"Code block {i+1} failed: {e}\nCode:\n{code}")
+            
+    assert 'LinearTransform' in namespace, "LinearTransform class not found"
+    assert 'PIDController' in namespace, "PIDController class not found"
+
+    # Test instantiation
+    lt = namespace['LinearTransform'](instance_name="test")
+    pc = namespace['PIDController'](instance_name="test")
+
+
+@pytest.mark.doc
+def test_model_exchange_examples():
+    """Test Model Exchange examples"""
+    test_dir = pathlib.Path(__file__).parent
+    mx_file = test_dir.parent.parent / "docs" / "usageMX.md"
+    
+    code_blocks = extract_python_code_from_md(mx_file)
+    
+    namespace = {}
+    exec("from pythonfmu3 import *", namespace)
+    
+    for i, code in enumerate(code_blocks):
+        try:
+            exec(code, namespace)
+        except Exception as e:
+            pytest.fail(f"MX Code block {i+1} failed: {e}")
+    
+    # Test that classes were actually created
+    assert 'VanDerPol' in namespace, "VanDerPol class not found"
+    assert 'BouncingBall' in namespace, "BouncingBall class not found"
+    
+    # Test instantiation
+    vdp = namespace['VanDerPol'](instance_name="test")
+    
+    bb = namespace['BouncingBall'](instance_name="test")

{pythonfmu3-0.3.2 → pythonfmu3-0.3.3/pythonfmu3.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pythonfmu3
-Version: 0.3.2
+Version: 0.3.3
 Summary: A lightweight framework that enables the packaging of Python3.x code as co-simulation FMUs.
 Home-page: https://github.com/stephensmith25/PythonFMU3
 Author: Stephen Smith
@@ -173,6 +173,7 @@ To create a model exchange FMU you must inherit from `Fmi3SlaveBase` and one of
 ```python
 
 from pythonfmu3 import Fmi3Causality, Fmi3SlaveBase, Float64, ModelExchange
+from typing import List
 
 
 class PythonSlaveMX(Fmi3SlaveBase, ModelExchange):

{pythonfmu3-0.3.2 → pythonfmu3-0.3.3}/pythonfmu3.egg-info/SOURCES.txt

@@ -47,6 +47,7 @@ pythonfmu3/tests/__init__.py
 pythonfmu3/tests/conftest.py
 pythonfmu3/tests/test_builder.py
 pythonfmu3/tests/test_deploy.py
+pythonfmu3/tests/test_docs.py
 pythonfmu3/tests/test_fmi3slave.py
 pythonfmu3/tests/test_integration.py
 pythonfmu3/tests/test_logger.py

{pythonfmu3-0.3.2 → pythonfmu3-0.3.3}/setup.cfg

@@ -49,6 +49,7 @@ tests =
 [tool:pytest]
 markers = 
 	integration: marks integration tests
+	doc: marks documentation tests
 
 [egg_info]
 tag_build = 

pythonfmu3-0.3.2/pythonfmu3/_version.py

@@ -1 +0,0 @@
-__version__ = "0.3.2"