def test_equal(self): unlabelled_leaf = RankTree(children=[]) self.assertEqual(unlabelled_leaf, unlabelled_leaf) self.assertTrue(unlabelled_leaf.shape_equal(unlabelled_leaf)) leaf_zero = RankTree(children=[], label=0) leaf_one = RankTree(children=[], label=1) leaf_two = RankTree(children=[], label=2) self.assertEqual(leaf_zero, leaf_zero) self.assertNotEqual(leaf_zero, leaf_one) self.assertTrue(leaf_zero.shape_equal(leaf_one)) tree1 = RankTree(children=[leaf_zero, leaf_one]) self.assertEqual(tree1, tree1) self.assertNotEqual(tree1, unlabelled_leaf) self.assertFalse(tree1.shape_equal(unlabelled_leaf)) tree2 = RankTree(children=[leaf_two, leaf_one]) self.assertNotEqual(tree1, tree2) self.assertTrue(tree1.shape_equal(tree2))
def test_equal(self): unlabelled_leaf = RankTree(children=[]) assert unlabelled_leaf == unlabelled_leaf assert unlabelled_leaf.shape_equal(unlabelled_leaf) leaf_zero = RankTree(children=[], label=0) leaf_one = RankTree(children=[], label=1) leaf_two = RankTree(children=[], label=2) assert leaf_zero == leaf_zero assert leaf_zero != leaf_one assert leaf_zero.shape_equal(leaf_one) tree1 = RankTree(children=[leaf_zero, leaf_one]) assert tree1 == tree1 assert tree1 != unlabelled_leaf assert not tree1.shape_equal(unlabelled_leaf) tree2 = RankTree(children=[leaf_two, leaf_one]) assert tree1 != tree2 assert tree1.shape_equal(tree2)