 }
 
 class DeviceLayoutConstraintSpy: DeviceLayoutConstraint {
+    var expectedDeviceSize: Size!
     var layoutIfNeededCalledCount: Int = 0
+    
     override open func layoutIfNeeded() {
         layoutIfNeededCalledCount += 1
     }
+    
+    override open func deviceSize() -> Size {
+        return expectedDeviceSize
+    }
 }

        describe("DeviceLayoutConstraint") {
            context("when use iPhone6") {
                beforeEach {
                    sut.deviceSize = .screen5_5Inch
                    sut.inch5_5 = Expected.Constant.inch5_5
                }
                


import XCTest

import Device
import Quick
import Nimble

