def test_windrose_mesh(): # Plot windrose epw_path = os.path.join(os.getcwd(), 'tests/fixtures/epw/tokyo.epw') epw = EPW(epw_path) w = WindRose(epw.wind_direction, epw.wind_speed, 16) w.legend_parameters.segment_count = 3 w.show_zeros = False w.show_stack = False mesh = w.colored_mesh # Simple test type assert isinstance(mesh, Mesh2D)
def test_simple_windrose_mesh(): # Testing vals dir_vals = [0, 0, 0, 10, 10, 10, 85, 90, 90, 90, 95, 170, 285, 288] spd_vals = dir_vals # Make into fake data collections a_per = AnalysisPeriod(6, 21, 12, 6, 21, 13) dates = [DateTime(6, 21, i) for i in range(len(dir_vals))] spd_header = Header(Speed(), 'm/s', a_per) dir_header = Header(GenericType('Direction', 'deg'), 'deg', a_per) spd_data = HourlyDiscontinuousCollection(spd_header, spd_vals, dates) dir_data = HourlyDiscontinuousCollection(dir_header, dir_vals, dates) # Init simple dir set divided by 4 w = WindRose(dir_data, spd_data, 4) w.legend_parameters.segment_count = 3 w.show_zeros = False w.show_stack = False mesh = w.colored_mesh assert isinstance(mesh, Mesh2D) # All true w = WindRose(dir_data, spd_data, 4) w.legend_parameters.segment_count = 10 w.show_zeros = True w.show_stack = True mesh = w.colored_mesh assert isinstance(mesh, Mesh2D) # Init simple dir set divided by 8 w = WindRose(dir_data, spd_data, 4) w.legend_parameters.segment_count = 3 w.show_zeros = False w.show_stack = False mesh = w.colored_mesh assert isinstance(mesh, Mesh2D)