def test_render_beached_elements(): """ Should this test be in map_canvas? """ input_file = os.path.join(data_dir, r"MapBounds_2Spillable2Islands2Lakes.bna") r = Renderer(input_file, output_dir, image_size=(800, 600)) BB = r.map_BB (min_lon, min_lat) = BB[0] (max_lon, max_lat) = BB[1] N = 100 # create some random particle positions: lon = random.uniform(min_lon, max_lon, (N, )) lat = random.uniform(min_lat, max_lat, (N, )) # create a sc sc = sample_sc_release(num_elements=N) sc['positions'][:, 0] = lon sc['positions'][:, 1] = lat # make half of them on land sc['status_codes'][::2] = oil_status.on_land r.create_foreground_image() r.draw_elements(sc) # create an uncertainty sc lon = random.uniform(min_lon, max_lon, (N, )) lat = random.uniform(min_lat, max_lat, (N, )) sc = sample_sc_release(num_elements=N, uncertain=True) sc['positions'][:, 0] = lon sc['positions'][:, 1] = lat # make half of them on land sc['status_codes'][::2] = oil_status.on_land r.draw_elements(sc) # save the image r.save_foreground(os.path.join(output_dir, 'foreground2.png')) assert True
def test_render_beached_elements(output_dir): """ Should this test be in map_canvas? """ r = Renderer(bna_sample, output_dir, image_size=(800, 600)) BB = r.map_BB (min_lon, min_lat) = BB[0] (max_lon, max_lat) = BB[1] N = 100 # create some random particle positions: lon = random.uniform(min_lon, max_lon, (N, )) lat = random.uniform(min_lat, max_lat, (N, )) # create a sc sc = sample_sc_release(num_elements=N) sc['positions'][:, 0] = lon sc['positions'][:, 1] = lat # make half of them on land sc['status_codes'][::2] = oil_status.on_land r.create_foreground_image() r.draw_elements(sc) # create an uncertainty sc lon = random.uniform(min_lon, max_lon, (N, )) lat = random.uniform(min_lat, max_lat, (N, )) sc = sample_sc_release(num_elements=N, uncertain=True) sc['positions'][:, 0] = lon sc['positions'][:, 1] = lat # make half of them on land sc['status_codes'][::2] = oil_status.on_land r.draw_elements(sc) # save the image r.save_foreground(os.path.join(output_dir, 'foreground2.png')) assert True
def test_render_elements(output_dir): """ Should this test be in map_canvas? """ # put in current dir for now: output_dir = './' r = Renderer(bna_sample, output_dir, image_size=(800, 600)) BB = r.map_BB (min_lon, min_lat) = BB[0] (max_lon, max_lat) = BB[1] N = 1000 # create some random particle positions: lon = random.uniform(min_lon, max_lon, (N, )) lat = random.uniform(min_lat, max_lat, (N, )) # create a sc sc = sample_sc_release(num_elements=N) sc['positions'][:, 0] = lon sc['positions'][:, 1] = lat r.create_foreground_image() r.draw_elements(sc) # create an uncertainty sc lon = random.uniform(min_lon, max_lon, (N, )) lat = random.uniform(min_lat, max_lat, (N, )) sc = sample_sc_release(num_elements=N, uncertain=True) sc['positions'][:, 0] = lon sc['positions'][:, 1] = lat r.draw_elements(sc) # save the image r.save_foreground(os.path.join(output_dir, 'foreground1.png')) assert True