def test_invalid_object(self): img = 'foo' attrs = get_source_img_attrs(img) self.assertEqual(attrs['height'], None) self.assertEqual(attrs['width'], None)
def test_invalid_object(self): img = "foo" attrs = get_source_img_attrs(img) self.assertEqual(attrs["height"], None) self.assertEqual(attrs["width"], None)
def test_valid_object(self): img = Mock() img.height = 100 img.width = 50 attrs = get_source_img_attrs(img) self.assertEqual(attrs['height'], 100) self.assertEqual(attrs['width'], 50)
def test_valid_object(self): img = Mock() img.height = 100 img.width = 50 attrs = get_source_img_attrs(img) self.assertEqual(attrs["height"], 100) self.assertEqual(attrs["width"], 50)