test_models.py

""" test models in the distribution
    this script finds all osim files in the repo, instantiate opensim.Model from each
    then call initSystem on it, roundtrips save/restore to compare.
"""
import os
import sys
import unittest
from fnmatch import fnmatch
import opensim
#import opensim as opensim
root = os.getcwd()
pattern = "*.osim"

osimpaths = list()
modelnames = list()

for path, subdirs, files in os.walk(root):
    for name in files:
        if fnmatch(name, pattern):
            osimpaths.append(os.path.join(path, name))
            modelnames.append(name)

for i in range(len(osimpaths)):
    print("\n\n\n" + 80 * "=" + "\nLoading model '%s'" % osimpaths[i] + "\n" + 80 * "-")
    # Without this next line, the print above does not necessarily
    # precede OpenSim's output.
    sys.stdout.flush()
    filename = osimpaths[i]
    modelname = modelnames[i]
    try:
        model = opensim.Model(filename)
        s = model.initSystem()
    except Exception as e:
        print("Oops, Model '%s' failed:\n%s" % (modelname, e.message))
        sys.exit(1)

    # Print the 4.0 version to file and then read back into memory
    filename_new = filename.replace('.osim','_new.osim')
    modelname_new = modelname.replace('.osim','_new.osim')
    # Print the 4.0 model to file
    model.printToXML(filename_new)
    # Try and read back in the file
    try:
        reloadedModel = opensim.Model(filename_new)
        s2 = reloadedModel.initSystem()
    except  Exception as e:
        print("Oops, 4.0 written Model '%s' failed:\n%s" % (modelname_new, e.message))
        sys.exit(1)

    # Remove the printed file
    os.remove(filename_new)

    if not reloadedModel.isEqualTo(model):
        print("Initial instance of '%s' is not equal to :\n%s" % (modelname,modelname_new))
        raise Exception("Compared two instances of 4.0 models are not equal")

print("All models loaded successfully.")