fixed issue with tagged strings

https://www.mikeash.com/pyblog/friday-qa-2015-07-31-tagged-pointer-strin gs.html
zengkebing3630 · Sep 21, 2015 · ef45369 · ef45369
1 parent a7c2e26
commit ef45369
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Source/DZNSegmentedControl.m b/Source/DZNSegmentedControl.m
@@ -429,11 +429,11 @@ - (void)setItems:(NSArray *)items
 
     id firstItem = [items firstObject];
 
-    NSPredicate *classPredicate = [NSPredicate predicateWithFormat:@"self isKindOfClass: %@", [firstItem class]];
-    NSAssert([items filteredArrayUsingPredicate:classPredicate].count == items.count, @"Cannot include different objects in the array. Please make sure to either pass an array of NSString or UIImage objects.");
-
     _imageMode = [firstItem isKindOfClass:[UIImage class]];
 
+    NSPredicate *classPredicate = [NSPredicate predicateWithFormat:@"self isKindOfClass: %@", _imageMode ? [UIImage class] : [NSString class]];
+    NSAssert([items filteredArrayUsingPredicate:classPredicate].count == items.count, @"Cannot include different objects in the array. Please make sure to either pass an array of NSString or UIImage objects.");
+
     _items = [NSArray arrayWithArray:items];
 
     if (!self.imageMode) {