def test_nndist_nn(self): nnd = s.nearest_neighbors(self.cell, num_neigh=2) self.assertEqual(nnd[0][0], 0) self.assertEqual(nnd[0][1], 29) self.assertEqual(nnd[1][0], 0) self.assertEqual(nnd[2][0], 1)
def test_nndist(self): nnd = s.nearest_neighbors(self.cell, tol=1.7) self.assertEqual(nnd[0][0], 6) self.assertEqual(nnd[0][1], 24) self.assertAlmostEqual(nnd[0][2], 1.6996583336037345)