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


import XCTest

import Quick
import Nimble


            
            context("when deviceSize is 4inch device") {
                beforeEach {
                    sut.expectedDeviceSize = .screen4Inch
                }
                
                context("when set constant per deviceSize") {
