//
//  TableViewControllerTests.swift
//  InlinePicker
//
//  Created by Jeffrey Kereakoglow on 4/21/16.
//  Copyright © 2016 Alexis Digital. All rights reserved.
//

import XCTest
@testable import InlinePicker

final class TableViewControllerTests: XCTestCase {
  private var viewController: TableViewController!

  override func setUp() {
    viewController = TableViewController(style: .Plain, dataSource: TableViewDataSource.dataSource)
  }

  override func tearDown() {
    viewController = nil
  }

  func testNumberOfSections() {
    XCTAssertEqual(viewController.numberOfSectionsInTableView(viewController.tableView), 1)
  }

  func testNumberOfRowsInSection() {
    XCTAssertEqual(
      viewController.tableView(viewController.tableView, numberOfRowsInSection: 0),
      TableViewDataSource.dataSource.count
    )
  }
}

//
//  TableViewControllerTests.swift
//  InlinePicker
//
//  Created by Jeffrey Kereakoglow on 4/21/16.
//  Copyright © 2016 Alexis Digital. All rights reserved.
//

import XCTest
@testable import InlinePicker

final class TableViewControllerTests: XCTestCase {
  private var tableVC: TableViewController!

  override func setUp() {
    tableVC = TableViewController(style: .Plain, dataSource: TableViewDataSource.dataSource)
  }

  override func tearDown() {
    tableVC = nil
  }

  func testDataSource() {
    XCTAssertNotNil(TableViewDataSource.dataSource)
  }
}

--- 
+++ 
@@ -10,17 +10,24 @@
 @testable import InlinePicker
 
 final class TableViewControllerTests: XCTestCase {
-  private var tableVC: TableViewController!
+  private var viewController: TableViewController!
 
   override func setUp() {
-    tableVC = TableViewController(style: .Plain, dataSource: TableViewDataSource.dataSource)
+    viewController = TableViewController(style: .Plain, dataSource: TableViewDataSource.dataSource)
   }
 
   override func tearDown() {
-    tableVC = nil
+    viewController = nil
   }
 
-  func testDataSource() {
-    XCTAssertNotNil(TableViewDataSource.dataSource)
+  func testNumberOfSections() {
+    XCTAssertEqual(viewController.numberOfSectionsInTableView(viewController.tableView), 1)
+  }
+
+  func testNumberOfRowsInSection() {
+    XCTAssertEqual(
+      viewController.tableView(viewController.tableView, numberOfRowsInSection: 0),
+      TableViewDataSource.dataSource.count
+    )
   }
 }
Add table view controller tests
