test/remove_cantilever_from_extension.py

   1 # Copyright (C) 2010-2012 W. Trevor King <wking@drexel.edu>
   2 #
   3 # This file is part of Hooke.
   4 #
   5 # Hooke is free software: you can redistribute it and/or modify it
   6 # under the terms of the GNU Lesser General Public License as
   7 # published by the Free Software Foundation, either version 3 of the
   8 # License, or (at your option) any later version.
   9 #
  10 # Hooke is distributed in the hope that it will be useful, but WITHOUT
  11 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  12 # or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General
  13 # Public License for more details.
  14 #
  15 # You should have received a copy of the GNU Lesser General Public
  16 # License along with Hooke.  If not, see
  17 # <http://www.gnu.org/licenses/>.
  18
  19 """
  20 >>> from hooke.hooke import Hooke, HookeRunner
  21 >>> h = Hooke()
  22 >>> r = HookeRunner()
  23 >>> h = r.run_lines(h, ['load_playlist test/data/test']) # doctest: +ELLIPSIS
  24 <FilePlaylist test.hkp>
  25 Success
  26 <BLANKLINE>
  27 >>> h = r.run_lines(h,
  28 ...     ['remove_cantilever_from_extension --distance_column "z piezo (m)" --deflection_column "deflection (m)"']
  29 ...     )
  30 Success
  31 <BLANKLINE>
  32 >>> curve = h.playlists.current().current()
  33 >>> approach = curve.data[0]
  34 >>> approach.info['columns']
  35 ['z piezo (m)', 'deflection (m)', 'cantilever adjusted extension (m)']
  36 >>> approach[:5,-1]  # doctest: +ELLIPSIS
  37 Data([ -1.806...e-06,  -1.812...e-06,  -1.817...e-06,
  38         -1.823...e-06,  -1.828...e-06])
  39 >>> approach[-5:,-1]  # doctest: +ELLIPSIS
  40 Data([ -1.981...e-06,  -1.986...e-06,  -1.980...e-06,
  41         -1.974...e-06,  -1.975...e-06])
  42
  43 The large shift is due to the unzeroed input deflection.
  44 """