如何在Swift

Question

extension UIScreen {
  static var width: CGFloat { return UIScreen.main.bounds.size.width }
  static var height: CGFloat { return UIScreen.main.bounds.size.height }
}

我想在单元测试中模拟这两个var，将宽度设置为414，高度：736，任何想法？