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))
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))