def test_item_is_self_similar(self):
sim_matrix, _ = self.out
diagonal = np.diagonal(sim_matrix)
self.assertEqual(diagonal.tolist(), [2.0, 1.0, 2.0, 2.0, 1.0])
微信
公众号
扫码关注公众号