def test_get_nearest_neighbors():
my_problem = NearestNeighborsOpt(schema)
result = my_problem.get_nearest_neighbors(5)
expected = np.array([[7, 9, 0, 3, 5],
[6, 7, 9, 1, 3],
[5, 7, 9, 2, 4],
[0, 5, 7, 1, 3],
[5, 7, 9, 2, 4],
[7, 0, 3, 9, 5],
[8, 9, 1, 3, 6],
[2, 4, 5, 3, 7],
[5, 6, 7, 8, 9],
[2, 3, 4, 5, 9]])
np.testing.assert_array_equal(result, expected, "Fail : get_nearest_neighbors")
