def test_visibility(self): document = Document('title1', 'desc1', 1, ['1.pdf', '2.pdf'], 'pdf') self.assertFalse(document.is_public()) document.make_public() self.assertTrue(document.is_public()) document.make_private() self.assertFalse(document.is_public())