fix

Author: muyexi

Demo/Base.lproj/Main.storyboard

@@ -28,7 +28,7 @@
         <!--View Controller-->
         <scene sceneID="nO0-P2-PU1">
             <objects>
-                <tableViewController id="zAi-6L-laa" customClass="ViewController" customModule="Demo" customModuleProvider="target" sceneMemberID="viewController">
+                <tableViewController storyboardIdentifier="ViewController" id="zAi-6L-laa" customClass="ViewController" customModule="Demo" customModuleProvider="target" sceneMemberID="viewController">
                     <tableView key="view" clipsSubviews="YES" contentMode="scaleToFill" alwaysBounceVertical="YES" dataMode="static" style="grouped" separatorStyle="default" rowHeight="44" sectionHeaderHeight="18" sectionFooterHeight="18" id="SfK-mt-6Qw">
                         <rect key="frame" x="0.0" y="0.0" width="375" height="667"/>
                         <autoresizingMask key="autoresizingMask" widthSizable="YES" heightSizable="YES"/>

DemoTests/DemoTests.swift

@@ -10,7 +10,8 @@ class DemoTests: XCTestCase {
         let storyboard = UIStoryboard(name: "Main", bundle: Bundle.main)
         let id = String(describing: ViewController.self)
 
-        viewController = storyboard.instantiateViewController(withIdentifier: id) as? ViewController
+        viewController = storyboard.instantiateViewController(withIdentifier: id) as! ViewController
+        viewController.viewDidLoad()
     }
 
     override func tearDown() {
@@ -30,11 +31,11 @@ class DemoTests: XCTestCase {
         XCTAssert(number == 1)
     }
 
-    func testHideSection() {
-        viewController.set(cells: viewController.hideMeCell3, viewController.hideMeCell4, hidden: false)
+    func testHideSectionRows() {
+        viewController.set(cells: viewController.hideMeCell3, viewController.hideMeCell4, hidden: true)
         viewController.reloadData(animated: true)
 
-        let number = viewController.tableView.numberOfRows(inSection: 1)
+        let number = viewController.tableView.numberOfRows(inSection: 2)
         XCTAssert(number == 0)
     }
 

Source/OriginalTable.swift

@@ -96,17 +96,22 @@ class OriginalTable {
         deleteIndexPaths.removeAll()
         updateIndexPaths.removeAll()
 
-        sections.flatMap { $0.rows }.forEach { (row) in
-            let indexPath = indexPathForInsertingOriginalRow(row)
-            
+        let allRows = sections.flatMap { $0.rows }
+        
+        allRows.forEach { (row) in
             if row.batchOperation == .delete {
+                let indexPath = indexPathForDeletingOriginalRow(row)
                 deleteIndexPaths.append(indexPath)
             } else if row.batchOperation == .insert {
+                let indexPath = indexPathForInsertingOriginalRow(row)
                 insertIndexPaths.append(indexPath)
             } else if row.batchOperation == .update {
+                let indexPath = indexPathForInsertingOriginalRow(row)
                 updateIndexPaths.append(indexPath)
             }
-            
+        }
+        
+        allRows.forEach { (row) in
             row.hiddenReal = row.hiddenPlanned
             row.batchOperation = BatchOperation.none
         }