def __init__(
self,
intervals,
intervals_unit,
elevations,
elevations_unit,
azimuths,
azimuths_unit,
distances,
distances_unit,
labels,
labels_unit,
confidence=None,
):
super().__init__(intervals, intervals_unit, labels, labels_unit, confidence)
annotations.validate_array_like(elevations, np.ndarray, float)
annotations.validate_array_like(azimuths, np.ndarray, float)
annotations.validate_array_like(distances, np.ndarray, float)
annotations.validate_lengths_equal(
[intervals, elevations, azimuths, distances, labels, confidence]
)
validate_locations(
np.concatenate(
[
elevations[:, np.newaxis],
azimuths[:, np.newaxis],
distances[:, np.newaxis],
],
axis=1,
)
)
annotations.validate_unit(elevations_unit, ELEVATIONS_UNITS)
annotations.validate_unit(azimuths_unit, AZIMUTHS_UNITS)
annotations.validate_unit(distances_unit, DISTANCES_UNITS)
self.elevations = elevations
self.azimuths = azimuths
self.distances = distances
self.elevations_unit = elevations_unit
self.azimuths_unit = azimuths_unit
self.distances_unit = distances_unit
def test_validate_array_like():
with pytest.raises(ValueError):
annotations.validate_array_like(None, list, str)
annotations.validate_array_like(None, list, str, none_allowed=True)
with pytest.raises(TypeError):
annotations.validate_array_like([1, 2], np.ndarray, str)
with pytest.raises(TypeError):
annotations.validate_array_like([1, 2], list, str)
with pytest.raises(TypeError):
annotations.validate_array_like(np.array([1, 2]), np.ndarray, str)
with pytest.raises(ValueError):
annotations.validate_array_like([], list, int)
def __init__(
self,
intervals,
intervals_unit,
time_step,
elevations,
elevations_unit,
azimuths,
azimuths_unit,
distances,
distances_unit,
labels,
labels_unit,
track_number_index,
confidence=None,
):
annotations.validate_array_like(intervals, list, list)
annotations.validate_array_like(labels, list, str)
annotations.validate_array_like(confidence,
np.ndarray,
float,
none_allowed=True)
[[
annotations.validate_intervals(intervals[np.newaxis, :])
for intervals in event_intervals
] for event_intervals in intervals]
annotations.validate_confidence(confidence)
annotations.validate_unit(labels_unit, LABEL_UNITS)
annotations.validate_unit(intervals_unit, TIME_UNITS)
self.intervals = intervals
self.intervals_unit = intervals_unit
self.labels = labels
self.labels_unit = labels_unit
self.confidence = confidence
annotations.validate_array_like(track_number_index, list, str)
annotations.validate_array_like(elevations, list, list)
annotations.validate_array_like(azimuths, list, list)
annotations.validate_array_like(distances, list, list)
annotations.validate_lengths_equal([
intervals,
elevations,
azimuths,
distances,
labels,
track_number_index,
confidence,
])
# validate location information for each event are numpy arrays
[[[
annotations.validate_array_like(subitem, np.ndarray, int)
for subitem in sitem
] for sitem in item] for item in [elevations, azimuths]]
[[
annotations.validate_array_like(subitem,
np.ndarray,
np.array([None]).dtype,
none_allowed=True)
for subitem in item
] for item in distances]
# validate length of location information is consistent
# for each event
[[
annotations.validate_lengths_equal([e, a, d])
for e, a, d in zip(els, azs, dis)
] for els, azs, dis in zip(elevations, azimuths, distances)]
[[
validate_locations(
np.concatenate(
[e[:, np.newaxis], a[:, np.newaxis], d[:, np.newaxis]],
axis=1)) for e, a, d in zip(els, azs, dis)
] for els, azs, dis in zip(elevations, azimuths, distances)]
[[
validate_time_steps(
time_step,
np.concatenate(
[e[:, np.newaxis], a[:, np.newaxis], d[:, np.newaxis]],
axis=1),
i,
) for e, a, d, i in zip(els, azs, dis, ivl)
] for els, azs, dis, ivl in zip(elevations, azimuths, distances,
intervals)]
annotations.validate_unit(elevations_unit, ELEVATIONS_UNITS)
annotations.validate_unit(azimuths_unit, AZIMUTHS_UNITS)
annotations.validate_unit(distances_unit, DISTANCES_UNITS)
self.time_step = time_step
self.elevations = elevations
self.azimuths = azimuths
self.distances = distances
self.track_number_index = track_number_index
self.elevations_unit = elevations_unit
self.azimuths_unit = azimuths_unit
self.distances_unit = distances_unit