Fix HeaderFormatter remove of style.

Author: SergioEstevao

Aztec/Classes/Formatters/Implementations/HeaderFormatter.swift

@@ -68,10 +68,11 @@ open class HeaderFormatter: ParagraphAttributeFormatter {
         resultingAttributes[.paragraphStyle] = newParagraphStyle
 
         if let font = attributes[.font] as? UIFont {
-            let newFont = font.withSize(CGFloat(header.defaultFontSize))
+            var newFont = font.withSize(CGFloat(header.defaultFontSize))
             if Configuration.headersWithBoldTrait {
-                resultingAttributes[.font] = newFont.modifyTraits(.traitBold, enable: false)
+                 newFont = newFont.modifyTraits(.traitBold, enable: false)
             }
+            resultingAttributes[.font] = newFont
         }
         resultingAttributes[.headingRepresentation] = nil
         return resultingAttributes

AztecTests/Formatters/HeaderFormatterTests.swift

@@ -28,7 +28,7 @@ class HeaderFormatterTests: XCTestCase {
 
         let removedAttrs = formatter.remove(from: updatedAttrs)
         let removedFont = removedAttrs[.font] as! UIFont
-        XCTAssert(removedFont.pointSize == defaultFontSize)
+        XCTAssertEqual(removedFont.pointSize, defaultFontSize)
     }
 
     /// Verifies that the Default Font is preserved whenever a Header Style is applied on top of an existant Header.
@@ -46,6 +46,6 @@ class HeaderFormatterTests: XCTestCase {
 
         let removedAttrs = formatterH2.remove(from: updatedH2Attrs)
         let removedFont = removedAttrs[.font] as! UIFont
-        XCTAssert(removedFont.pointSize == defaultFontSize)
+        XCTAssertEqual(removedFont.pointSize, defaultFontSize)
     }
 }