import numpy as np
import matplotlib.pyplot as plt
import astropy.units as u
from astropy.timeseries import TimeSeries
from swxsoc.swxdata import SWXData
bx = np.concatenate([[0], np.random.choice(a=[-1, 0, 1], size=1000)]).cumsum(0)
by = np.concatenate([[0], np.random.choice(a=[-1, 0, 1], size=1000)]).cumsum(0)
bz = np.concatenate([[0], np.random.choice(a=[-1, 0, 1], size=1000)]).cumsum(0)
ts = TimeSeries(time_start="2016-03-22T12:30:31", time_delta=3 * u.s, data={"Bx": u.Quantity(bx, "nanoTesla", dtype=np.int16)})
input_attrs = SWXData.global_attribute_template("nemisis", "l1", "1.0.0")
sw_data = SWXData(timeseries=ts, meta=input_attrs)
sw_data.add_measurement(measure_name=f"By", data=u.Quantity(by, 'nanoTesla', dtype=np.int16))
sw_data.add_measurement(measure_name=f"Bz", data=u.Quantity(bz, 'nanoTesla', dtype=np.int16))
fig = plt.figure()
sw_data.plot() # doctest: +SKIP
plt.show() # doctest: +SKIP