test/export_block.py

   1 # Copyright (C) 2010 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 >>> import os
  21 >>> import os.path
  22 >>> from uuid import uuid4
  23 >>> from hooke.hooke import Hooke, HookeRunner
  24
  25 >>> h = Hooke()
  26 >>> r = HookeRunner()
  27 >>> h = r.run_lines(h, ['load_playlist test/data/test']) # doctest: +ELLIPSIS
  28 <FilePlaylist test.hkp>
  29 Success
  30 <BLANKLINE>
  31 >>> file_name = '%s.dat' % uuid4()
  32 >>> export_already_exists = os.path.exists(file_name)
  33 >>> export_already_exists
  34 False
  35 >>> h = r.run_lines(h, ['export_block --output %s' % file_name])
  36 Success
  37 <BLANKLINE>
  38 >>> with open(file_name, 'r') as f:
  39 ...     lines = f.readlines()
  40 >>> if export_already_exists == False:
  41 ...    os.remove(file_name)
  42 >>> print len(lines)
  43 2049
  44 >>> print ''.join(lines[:5]),  # doctest: +ELLIPSIS, +REPORT_UDIFF +NORMALIZE_WHITESPACE
  45 # z piezo (m)  deflection (m)
  46 -1.519...e-07  9.094...e-08
  47 -1.513...e-07  9.130...e-08
  48 -1.513...e-07  9.157...e-08
  49 -1.507...e-07  9.189...e-08
  50 """