def mock_get_input_datasets(data_sources, t0, dt, n_steps): for data in data_sources: connector = data.pop("connector") # We don't have a connector, so ignore here. for station_id, sources in data.iteritems(): for source, uid in sources.iteritems(): if source == "radiation": mock_data = create_mock_station_data(t0, dt, n_steps)
def mock_station_data(bounding_box, n_x, n_y): geo_positions = [] x_min, x_max, y_min, y_max = bounding_box dx = float(x_max - x_min)/n_x dy = float(y_max - y_min)/n_y data = [] for i in xrange(n_x): x = x_min + (i + 0.5)*dx for j in xrange(n_y): y = y_min + (j + 0.5)*dy z = abs(1000*math.sin(x/10000.0 + y/10000.0)) # Crop at 1000 mas geo_positions.append([x, y, z]) data.append(InputSource([x, y, z], create_mock_station_data(0, 3600, 10))) return data