python source code of input_element_int_test

Project: python-control (GitHub Link)

python-control-master
- ChangeLog
- conda-recipe
  - meta.yaml
  - bld.bat
- .coveragerc
- examples
  - secord-matlab.py
  - steering-gainsched.py
  - pvtol-lqr.py
  - pvtol-nested-ss.py
  - type2_type3.py
  - run_examples.sh
  - cruise-control.py
  - robust_mimo.py
  - README
  - rss-balred.py
  - pvtol-lqr-nested.ipynb
  - slycot-import-test.py
  - bdalg-matlab.py
  - pvtol-nested.py
  - robust_siso.py
  - check-controllability-and-observability.py
  - phaseplots.py
  - kincar-flatsys.py
  - genswitch.py
  - tfvis.py
  - test-response.py
- LICENSE
- doc-requirements.txt
- external
  - controls.py
  - yottalab.py
- README.rst
- setup.py
- runtests.py
- make_version.py
- setup.cfg
- .travis.yml
- Pending
- .gitignore
- MANIFEST.in
- control
  - margins.py
  - modelsimp.py
  - rlocus.py
  - statefbk.py
  - frdata.py
  - flatsys
    - systraj.py
    - basis.py
    - linflat.py
    - flatsys.py
    - poly.py
    - __init__.py
  - matlab
    - wrappers.py
    - __init__.py
    - timeresp.py
  - config.py
  - lti.py
  - nichols.py
  - iosys.py
  - mateqn.py
  - statesp.py
  - bdalg.py
  - grid.py
  - dtime.py
  - pzmap.py
  - setup.py
  - robust.py
  - canonical.py
  - delay.py
  - __init__.py
  - sisotool.py
  - exception.py
  - phaseplot.py
  - tests
    - input_element_int_test.py
    - freqresp_test.py
    - lti_test.py
    - nichols_test.py
    - test_control_matlab.py
    - statefbk_array_test.py
    - rlocus_test.py
    - sisotool_test.py
    - minreal_test.py
    - slycot_convert_test.py
    - frd_test.py
    - matlab_test.py
    - convert_test.py
    - statesp_array_test.py
    - delay_test.py
    - mateqn_test.py
    - ctrlutil_test.py
    - flatsys_test.py
    - discrete_test.py
    - canonical_test.py
    - robust_array_test.py
    - statesp_test.py
    - modelsimp_array_test.py
    - xferfcn_input_test.py
    - robust_test.py
    - __init__.py
    - margin_test.py
    - iosys_test.py
    - xferfcn_test.py
    - statefbk_test.py
    - bdalg_test.py
    - config_test.py
    - modelsimp_test.py
    - phaseplot_test.py
    - timeresp_test.py
  - timeresp.py
  - bench
    - time_freqresp.py
  - ctrlutil.py
  - freqplot.py
  - xferfcn.py
- doc
  - secord-matlab.py
  - steering-gainsched.py
  - Makefile
  - classes.rst
  - secord-matlab.rst
  - flatsys.rst
  - robust_mimo.rst
  - pvtol-lqr.py
  - robust_siso.rst
  - intro.rst
  - kincar-flatsys.rst
  - cruise-control.py
  - conventions.rst
  - iosys.rst
  - robust_mimo.py
  - README
  - rss-balred.py
  - pvtol-lqr-nested.ipynb
  - control.rst
  - pvtol-nested.py
  - examples.rst
  - steering.ipynb
  - robust_siso.py
  - phaseplots.rst
  - cruise.ipynb
  - phaseplots.py
  - steering-gainsched.rst
  - pvtol-nested.rst
  - cruise-control.rst
  - index.rst
  - conf.py
  - kincar-flatsys.py
  - rss-balred.rst
  - pvtol-lqr.rst
  - matlab.rst

# input_element_int_test.py
#
# Author: Kangwon Lee (kangwonlee)
# Date: 22 Oct 2017
#
# Unit tests contributed as part of PR #158, "SISO tf() may not work
# with numpy arrays with numpy.int elements"
#
# Modified:
# * 29 Dec 2017, RMM - updated file name and added header

import unittest
import numpy as np
import control as ctl

class TestTfInputIntElement(unittest.TestCase):
    # currently these do not pass
    def test_tf_den_with_numpy_int_element(self):
        num = 1
        den = np.convolve([1, 2, 1], [1, 1, 1])

        sys = ctl.tf(num, den)

        self.assertAlmostEqual(1.0, ctl.dcgain(sys))

    def test_tf_num_with_numpy_int_element(self):
        num = np.convolve([1], [1, 1])
        den = np.convolve([1, 2, 1], [1, 1, 1])

        sys = ctl.tf(num, den)

        self.assertAlmostEqual(1.0, ctl.dcgain(sys))

    # currently these pass
    def test_tf_input_with_int_element_works(self):
        num = 1
        den = np.convolve([1.0, 2, 1], [1, 1, 1])

        sys = ctl.tf(num, den)

        self.assertAlmostEqual(1.0, ctl.dcgain(sys))

    def test_ss_input_with_int_element(self):
        ident = np.matrix(np.identity(2), dtype=int)
        a = np.matrix([[0, 1],
                       [-1, -2]], dtype=int) * ident
        b = np.matrix([[0],
                       [1]], dtype=int)
        c = np.matrix([[0, 1]], dtype=int)
        d = 0

        sys = ctl.ss(a, b, c, d)
        sys2 = ctl.ss2tf(sys)
        self.assertAlmostEqual(ctl.dcgain(sys), ctl.dcgain(sys2))