예제 #1
0
파일: test.py 프로젝트: kzvezdarov/PyCEbox 
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()
예제 #2
0
파일: test.py 프로젝트: kzvezdarov/PyCEbox 
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()
예제 #3
0
파일: test.py 프로젝트: kzvezdarov/PyCEbox 
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()
예제 #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()
예제 #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()
예제 #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()