def test_support_101_plot_API(input_values, expected_values):
# Define expected values.
expected_plot_axis_indices, expected_axes_coordinates, expected_axes_units, \
expected_data_unit, expected_kwargs = expected_values
# Run function
output_plot_axis_indices, output_axes_coordinates, output_axes_units, \
output_data_unit, output_kwargs = plotting._support_101_plot_API(*input_values)
# Check values are correct
assert output_plot_axis_indices == expected_plot_axis_indices
if expected_axes_coordinates is None:
assert output_axes_coordinates == expected_axes_coordinates
elif type(expected_axes_coordinates) is list:
for i, ac in enumerate(output_axes_coordinates):
np.testing.assert_array_equal(ac, expected_axes_coordinates[i])
assert output_axes_units == expected_axes_units
assert output_data_unit == expected_data_unit
assert output_kwargs == expected_kwargs
def test_support_101_plot_API(input_values, expected_values):
# Define expected values.
expected_plot_axis_indices, expected_axes_coordinates, expected_axes_units, \
expected_data_unit, expected_kwargs = expected_values
# Run function
output_plot_axis_indices, output_axes_coordinates, output_axes_units, \
output_data_unit, output_kwargs = plotting._support_101_plot_API(*input_values)
# Check values are correct
assert output_plot_axis_indices == expected_plot_axis_indices
if expected_axes_coordinates is None:
assert output_axes_coordinates == expected_axes_coordinates
elif type(expected_axes_coordinates) is list:
for i, ac in enumerate(output_axes_coordinates):
np.testing.assert_array_equal(ac, expected_axes_coordinates[i])
assert output_axes_units == expected_axes_units
assert output_data_unit == expected_data_unit
assert output_kwargs == expected_kwargs
def test_support_101_plot_API_errors(input_values):
with pytest.raises(ValueError):
output = plotting._support_101_plot_API(*input_values)
def test_support_101_plot_API_errors(input_values):
with pytest.raises(ValueError):
output = plotting._support_101_plot_API(*input_values)
