add test

Sources/DeviceModel.swift

@@ -21,7 +21,7 @@
 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 // SOFTWARE.
 
-public enum DeviceModel {
+public enum DeviceModel: CaseIterable {
 
     case iPhone4, iPhone4S
     case iPhone5, iPhone5C, iPhone5S
@@ -201,3 +201,17 @@ extension DeviceModel {
         }
     }
 }
+
+// MARK: Detecting the Notch
+
+extension DeviceModel {
+    public var hasNotch: Bool {
+        switch self {
+
+        case .iPhoneX, .iPhoneXS, .iPhoneXSMax, .iPhoneXR:
+            return true
+        default:
+          return false
+        }
+    }
+}

Sources/UIDeviceExtensions.swift

@@ -67,13 +67,7 @@ public extension UIDeviceComplete where Base == UIDevice {
     }
 
     var hasNotch: Bool {
-      switch deviceModel {
-
-      case .iPhoneX, .iPhoneXS, .iPhoneXSMax, .iPhoneXR:
-        return true
-      default:
-        return false
-      }
+        return deviceModel.hasNotch
     }
 }
 

Tests/UIDeviceExtensionsTests.swift

@@ -55,5 +55,14 @@ class UIDeviceExtensionsTests: XCTestCase {
         let deviceFamily = DeviceFamily(rawValue: "x86_64")
         XCTAssert(deviceFamily == .simulator, "DeviceExtensions - .isSimulator is failing")
     }
-
+
+
+    func testHasNotch() {
+      let notchModels: [DeviceModel] = [.iPhoneX, .iPhoneXS, .iPhoneXSMax, .iPhoneXR]
+
+      let noNotchModels: [DeviceModel] = DeviceModel.allCases.filter( { !notchModels.contains($0) })
+
+      notchModels.forEach { XCTAssertTrue($0.hasNotch) }
+      noNotchModels.forEach { XCTAssertFalse($0.hasNotch) }
+    }
 }