Changeset 5899 for anuga_core/source_numpy_conversion/anuga/abstract_2d_finite_volumes/test_ermapper.py
- Timestamp:
- Nov 6, 2008, 12:28:22 PM (15 years ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
anuga_core/source_numpy_conversion/anuga/abstract_2d_finite_volumes/test_ermapper.py
r1813 r5899 1 ## Automatically adapted for numpy.oldnumeric Oct 28, 2008 by alter_code1.py 2 1 3 #!/usr/bin/env python 2 4 … … 5 7 6 8 import ermapper_grids 7 import Numeric 9 ##import numpy.oldnumeric as Numeric 10 import numpy 8 11 from os import remove 9 12 … … 19 22 data_filename = 'test_write_ermapper_grid' 20 23 21 original_grid = Numeric.array([[0.0, 0.1, 1.0], [2.0, 3.0, 4.0]])24 original_grid = numpy.array([[0.0, 0.1, 1.0], [2.0, 3.0, 4.0]]) 22 25 23 26 # Check that the function works when passing the filename without … … 25 28 ermapper_grids.write_ermapper_grid(data_filename, original_grid) 26 29 new_grid = ermapper_grids.read_ermapper_grid(data_filename) 27 assert Numeric.allclose(original_grid, new_grid)30 assert numpy.allclose(original_grid, new_grid) 28 31 29 32 # Check that the function works when passing the filename with … … 31 34 ermapper_grids.write_ermapper_grid(header_filename, original_grid) 32 35 new_grid = ermapper_grids.read_ermapper_grid(header_filename) 33 assert Numeric.allclose(original_grid, new_grid)36 assert numpy.allclose(original_grid, new_grid) 34 37 35 38 # Clean up created files … … 40 43 # Setup test data 41 44 filename = 'test_write_ermapper_grid' 42 original_grid = Numeric.array([0.0, 0.1, 1.0, 2.0, 3.0, 4.0])45 original_grid = numpy.array([0.0, 0.1, 1.0, 2.0, 3.0, 4.0]) 43 46 44 47 # Write test data 45 ermapper_grids.write_ermapper_data(original_grid, filename, Numeric.Float64)48 ermapper_grids.write_ermapper_data(original_grid, filename, numpy.float64) 46 49 47 50 # Read in the test data 48 new_grid = ermapper_grids.read_ermapper_data(filename, Numeric.Float64)51 new_grid = ermapper_grids.read_ermapper_data(filename, numpy.float64) 49 52 50 53 # Check that the test data that has been read in matches the original data 51 assert Numeric.allclose(original_grid, new_grid)54 assert numpy.allclose(original_grid, new_grid) 52 55 53 56 # Clean up created files … … 57 60 # Setup test data 58 61 filename = 'test_write_ermapper_grid' 59 original_grid = Numeric.array([0.0, 0.1, 1.0, 2.0, 3.0, 4.0])62 original_grid = numpy.array([0.0, 0.1, 1.0, 2.0, 3.0, 4.0]) 60 63 61 64 # Write test data … … 66 69 67 70 # Check that the test data that has been read in matches the original data 68 assert Numeric.allclose(original_grid, new_grid)71 assert numpy.allclose(original_grid, new_grid) 69 72 70 73 # Clean up created files … … 75 78 76 79 # setup test data 77 original_grid = Numeric.array([[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]])80 original_grid = numpy.array([[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]]) 78 81 # Write test data 79 82 ermapper_grids.write_ermapper_data(original_grid, data_filename)
Note: See TracChangeset
for help on using the changeset viewer.