def test_containing_raises_on_no_pkg_contains_part(self): """_Package.containing(part) raises on no package contains part""" # setup ------------------------ pkg = _Package(test_pptx_path) pkg.presentation # does nothing, just needed to fake out pep8 warning part = Mock(name='part') # verify ----------------------- with self.assertRaises(KeyError): _Package.containing(part)
def test_containing_returns_correct_pkg(self): """_Package.containing() returns right package instance""" # setup ------------------------ pkg1 = _Package(test_pptx_path) pkg1.presentation # does nothing, just needed to fake out pep8 warning pkg2 = _Package(test_pptx_path) slide = pkg2.presentation.slides[0] # exercise --------------------- found_pkg = _Package.containing(slide) # verify ----------------------- expected = pkg2 actual = found_pkg msg = "expected %r, got %r" % (expected, actual) self.assertEqual(expected, actual, msg)