 import UIKit
 import XCTest
 import SwiftPhotoGallery
-import Nimble
 
 class SwiftPhotoGalleryTests: XCTestCase {
 

        let indexPath = IndexPath(item: 0, section: 0)
        testGallery.imageCollectionView.reloadItems(at: [indexPath])

        expect(self.testHelper.timesAskedForImageInGallery[0]).to(beGreaterThan(0))
        expect(self.testHelper.timesAskedForImageInGallery[1]).to(beGreaterThan(0))
    }

    func testSetDataSourceReloadsImages() {
        let newDataSource = SwiftPhotoGalleryTestHelper()

        expect(newDataSource.timesAskedForNumberOfImagesInGallery).to(equal(0))

        testGallery.dataSource = newDataSource



        let timesAskedForNumberOfImagesInGallery = testHelper.timesAskedForNumberOfImagesInGallery

        XCTAssert(self.testHelper.timesAskedForNumberOfImagesInGallery == 4)

        testGallery.dataSource = testHelper

        XCTAssert(self.testHelper.timesAskedForNumberOfImagesInGallery == timesAskedForNumberOfImagesInGallery)
    }

    func testTapCallsDelegateMethod() {
