Merge pull request #84 from qasimsina/master

Update FoldingCell.swift

FoldingCell.podspec

@@ -1,6 +1,6 @@
 Pod::Spec.new do |s|
   s.name         = 'FoldingCell'
-  s.version      = '2.0.1'
+  s.version      = '2.0.2'
   s.summary      = 'UITableViewCell with folding animation.'
   s.homepage     = 'https://github.com/Ramotion/folding-cell'
   s.license      = 'MIT'

FoldingCell/FoldingCell/FoldingCell/FoldingCell.swift

@@ -217,9 +217,13 @@ open class FoldingCell: UITableViewCell {
     let itemHeight = (contSize.height - 2 * forgSize.height) / CGFloat(itemCount - 2)
     
     if itemCount == 2 {
-      // decrease containerView height or increase itemCount
-      assert(contSize.height - 2 * forgSize.height == 0, "contanerView.height too high")
-    }
+            // decrease containerView height or increase itemCount
+            assert(contSize.height - 2 * forgSize.height == 0, "contanerView.height too high")
+        }
+        else{
+            // decrease containerView height or increase itemCount
+            assert(contSize.height - 2 * forgSize.height >= itemHeight, "contanerView.height too high")
+        }
     // decrease containerView height or increase itemCount
     assert(contSize.height - 2 * forgSize.height >= itemHeight, "contanerView.height too high")
     

README.md

@@ -28,10 +28,7 @@ The [iPhone mockup](https://store.ramotion.com/product/iphone-6-mockups?utm_sour
 Just add the FoldingCell.swift file to your project.
 
 or use [CocoaPods](https://cocoapods.org) with Podfile:
-``` ruby
-pod 'FoldingCell' '~> 1.0.0' swift 2.2
-pod 'FoldingCell' '~> 2.0.1' swift 3
-```
+pod 'FoldingCell' '~> 2.0.2' 
 
 or [Carthage](https://github.com/Carthage/Carthage) users can simply add Mantle to their `Cartfile`:
 ```