class TestSortedSetAdd(unittest.TestCase): def setUp(self): self.ss = SortedSet() def test_add(self): self.ss.add(3) self.assertEqual(len(self.ss), 1) def test_add_with_an_elem_already_in_the_set(self): self.ss.add(3) self.ss.add(3) self.assertEqual(len(self.ss), 1) @unittest.skipIf(PY2, "comparing different types is ok in PY2") def test_raise_unorderable_types(self): self.ss.add(3) self.assertRaises(TypeError, self.ss.add, 'sepeth')