def test_write_default_header(self): data_filename = 'test_write_ermapper_grid' # setup test data original_grid = num.array([[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]]) # Write test data ermapper_grids.write_ermapper_data(original_grid, data_filename) # Write test header using all default values header_filename = data_filename + '.ers' ermapper_grids.write_ermapper_header(header_filename) # Check that the read in values match the default values header = ermapper_grids.read_ermapper_header(header_filename) assert header['datum'] == '"GDA94"' assert header['projection'] == '"GEOGRAPHIC"' assert header['coordinatetype'] == 'LL' assert header['rotation'] == '0:0:0.0' assert header['celltype'] == 'IEEE4ByteReal' assert header['nullcellvalue'] == '-99999' assert header['xdimension'] == '100' assert header['registrationcellx'] == '0' assert header['ydimension'] == '100' assert header['registrationcelly'] == '2' assert header['nroflines'] == '3' assert header['nrofcellsperline'] == '4' assert header['longitude'] == '0:0:0' assert header['latitude'] == '0:0:0' assert header['nrofbands'] == '1' assert header['value'] == '"Default_Band"' # Clean up created files remove(data_filename) remove(header_filename)
def test_basic_single_line_grid_default_format(self): # Setup test data filename = 'test_write_ermapper_grid' original_grid = num.array([0.0, 0.1, 1.0, 2.0, 3.0, 4.0]) # Write test data ermapper_grids.write_ermapper_data(original_grid, filename) # Read in the test data new_grid = ermapper_grids.read_ermapper_data(filename) # Check that the test data that has been read in matches the original data assert num.allclose(original_grid, new_grid) # Clean up created files remove(filename)