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


import XCTest

import Quick
import Nimble


}

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