HaversineDistance extends FastDistance { public double calc(double latitude1, double longitude1, double latitude2, double longitude2) { double lat1 = degreeToRadian(latitude1); double lon1 = degreeToRadian(longitude1); double lat2 = degreeToRadian(latitude2); double lon2 = degreeToRadian(longitude2); double vLat = lat2 - lat1; double vLon = lon2 - lon1; double h = haverSine(vLat) + Math.cos(lat1) * Math.cos(lat2) * haverSine(vLon); return cal(h); }  double calc(double latitude1, double longitude1, double latitude2, double longitude2); @Override double calc(double[] array1, double[] array2); @Override double calc(Vector vec1, Vector vec2); @Override void updateLabel(FastDistanceData data);  }
@Test public void testContinuousDistance() { Assert.assertEquals(distance.calc(denseVector1, denseVector2), 5160.251, 0.01); Assert.assertEquals(distance.calc(denseVector1.getData(), denseVector2.getData()), 5160.251, 0.01); }
@Test public void testCalDistanceVecVec() { DenseMatrix denseResult = distance.calc(distance.prepareVectorData(Tuple2.of(denseVector1, null)), (FastDistanceData)distance.prepareVectorData(Tuple2.of(denseVector2, null))); Assert.assertEquals(denseResult.get(0, 0), 5160.251, 0.01); }
@Test public void testCalDistanceMatrixMatrix() { FastDistanceMatrixData matrixData = initMatrixData(); DenseMatrix res = distance.calc(matrixData, matrixData); double[] expect = new double[] {0.0, 5160.251, 5160.251, 0.0}; double[] predict = res.getData(); for (int i = 0; i < expect.length; i++) { Assert.assertEquals(expect[i], predict[i], 0.01); } }
@Test public void testCalDistanceVecMatrix() { FastDistanceMatrixData matrixData = initMatrixData(); FastDistanceVectorData vectorData = distance.prepareVectorData(Row.of(0, denseVector1), 1); double[] predict = distance.calc(matrixData, vectorData).getData(); double[] expect = new double[] {0.0, 5160.251}; for (int i = 0; i < expect.length; i++) { Assert.assertEquals(expect[i], predict[i], 0.01); } predict = distance.calc(vectorData, matrixData).getData(); for (int i = 0; i < expect.length; i++) { Assert.assertEquals(expect[i], predict[i], 0.01); } }