Altered float check to also check for double

supermarin · Jun 17, 2015 · 5b16639 · 5b16639
1 parent a3e7870
commit 5b16639
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Classes/NSManagedObject+ActiveRecord.m b/Classes/NSManagedObject+ActiveRecord.m
@@ -372,7 +372,7 @@ - (void)setSafeValue:(id)value forKey:(NSString *)key {
         else if (attributeType == NSBooleanAttributeType)
             value = [NSNumber numberWithBool:[value boolValue]];
 
-        else if (attributeType == NSFloatAttributeType)
+        else if ([self isFloatAttributeType:attributeType])
             value = [NSNumber numberWithDouble:[value doubleValue]];
 
         else if (attributeType == NSDateAttributeType)
@@ -388,6 +388,11 @@ - (BOOL)isIntegerAttributeType:(NSAttributeType)attributeType {
            (attributeType == NSInteger64AttributeType);
 }
 
+- (BOOL)isFloatAttributeType:(NSAttributeType)attributeType {
+    return (attributeType == NSFloatAttributeType) ||
+           (attributeType == NSDoubleAttributeType);
+}
+
 #pragma mark - Date Formatting
 
 - (NSDateFormatter *)defaultFormatter {