         XCTAssertEqual(imageAttachment.extraAttributes["width"], nil)
     }
     
-    func testHeight() {
+    func testWidthGetter() {
+        let imageAttachment = ImageAttachment(identifier: "testing")
+        let width = 500
+        
+        XCTAssertEqual(imageAttachment.width, nil)
+        
+        imageAttachment.extraAttributes["width"] = String(width)
+        XCTAssertEqual(imageAttachment.width, width)
+        
+        imageAttachment.extraAttributes["width"] = nil
+        XCTAssertEqual(imageAttachment.width, nil)
+    }
+    
+    func testHeightSetter() {
         let imageAttachment = ImageAttachment(identifier: "testing")
         let height = 500
         

    
    func testImageIDChangesWhenExtraAttributeChanges() {
        let imageAttachment = ImageAttachment(identifier: "testing")
        
        imageAttachment.extraAttributes["class"] = "wp-image-200"
        XCTAssertEqual(imageAttachment.imageID, 200)

    func testImageIDChangesExtraAttributes() {
        let imageAttachment = ImageAttachment(identifier: "testing")
        
        imageAttachment.extraAttributes["class"] = "some-attributes some-more-attributes"
        
        imageAttachment.imageID = 200
