def test_get_reducer(option, expected_array): """Tests for ValueError when using an unsupported option.""" complex_array = np.array([1.0, 1.0+10j, 2.0, 2.0+20j], dtype='complex') func = get_reducer(option) real_array = func(complex_array) assert np.all(np.isreal(real_array)) assert_allclose(real_array, expected_array) # nothing should happen to an array that only contains real data assert_allclose(func(real_array), real_array)
def test_get_reducer_invalid_option(): """Tests for ValueError when using an unsupported option.""" option = 'unknown' msg = r'Invalid option' with pytest.raises(ValueError, match=msg): get_reducer(option)