Skip to content

Commit

Permalink
Created prototype for unittesting ONNX
Browse files Browse the repository at this point in the history
  • Loading branch information
Felix-Mac committed Oct 10, 2022
1 parent 31bfde9 commit b7bdd57
Showing 1 changed file with 64 additions and 0 deletions.
64 changes: 64 additions & 0 deletions testing/test_onnx.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,64 @@
#
# This file is part of do-mpc
#
# do-mpc: An environment for the easy, modular and efficient implementation of
# robust nonlinear model predictive control
#
# Copyright (c) 2014-2019 Sergio Lucia, Alexandru Tatulea-Codrean
# TU Dortmund. All rights reserved
#
# do-mpc is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation, either version 3
# of the License, or (at your option) any later version.
#
# do-mpc is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with do-mpc. If not, see <http://www.gnu.org/licenses/>.

import numpy as np
import matplotlib.pyplot as plt
from casadi import *
from casadi.tools import *
import pdb
import sys
import unittest
import pickle

from importlib import reload
import copy

do_mpc_path = '../'
if not do_mpc_path in sys.path:
sys.path.append('../')

import do_mpc




class TestONNX(unittest.TestCase):
def setUp(self):
"""Add path of test case and import the modules.
If this test isn't the first to run, the modules need to be reloaded.
Reset path afterwards.
"""
pass

def test_ONNX1(self):
print('Testing ONNX1 model')
# Load first onnx model
# onnx =

self.assertTrue(True)





if __name__ == '__main__':
unittest.main()

0 comments on commit b7bdd57

Please sign in to comment.