Exemplo n.º 1
0
def test_get_grid_points_num_grid_points_None(l):
    x = pd.Series(l)

    assert compare_with_NaN(x.unique(), ice.get_grid_points(x, None)).all()
Exemplo n.º 2
0
def test_get_grid_points_num_grid_points_too_many():
    x = pd.Series(np.array([0, 0, 1]))
    expected_grid_points = np.array([0, 0.5, 1])

    assert (expected_grid_points == ice.get_grid_points(x, 5)).all()
Exemplo n.º 3
0
def test_get_grid_points_num_grid_points():
    x = pd.Series(np.array([0, 0, 1, 2, 3, 4, 5, 6, 7]))
    expected_grid_points = np.array([0, 1, 3, 5, 7])

    assert (expected_grid_points == ice.get_grid_points(x, 5)).all()
Exemplo n.º 4
0
def test_get_grid_points_num_grid_points_None(l):
    x = pd.Series(l)

    assert compare_with_NaN(x.unique(), ice.get_grid_points(x, None)).all()
Exemplo n.º 5
0
def test_get_grid_points_num_grid_points_too_many():
    x = pd.Series(np.array([0, 0, 1]))
    expected_grid_points = np.array([0, 0.5, 1])

    assert (expected_grid_points == ice.get_grid_points(x, 5)).all()
Exemplo n.º 6
0
def test_get_grid_points_num_grid_points():
    x = pd.Series(np.array([0, 0, 1, 2, 3, 4, 5, 6, 7]))
    expected_grid_points = np.array([0, 1, 3, 5, 7])

    assert (expected_grid_points == ice.get_grid_points(x, 5)).all()