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)
