Ejemplo n.º 1
0
 def test_width_one(self):
     expected = matrix([[0, 1, 10, 10], [1, 0, 1, 10], [10, 1, 0, 1], [10, 10, 1, 0]])
     dist = _build_distance_matrix(self.df, width=1, default_dist_value=10)
     result = dist == expected
     self.assertTrue(result.all(), "\n{0} \n!=\n {1}".format(expected, dist))
Ejemplo n.º 2
0
 def test_default_params(self):
     expected = matrix([[0, 1, 2, 3], [1, 0, 1, 2], [2, 1, 0, 1], [3, 2, 1, 0]])
     dist = _build_distance_matrix(self.df)
     result = dist == expected
     self.assertTrue(result.all(), "\n{0} \n!=\n {1}".format(expected, dist))