#import "FunctionalUtilTest.h"
#import "FunctionalUtil.h"
#import "TestUtil.h"

@implementation FunctionalUtilTest
-(void) testAny {
    test(![@[] any:^(id x) { return false; }]);
    test(![@[] any:^(id x) { return true; }]);
    test(![@[@1] any:^(id x) { return false; }]);
    test([@[@1] any:^(id x) { return true; }]);
    
    test([(@[@2, @3, @5]) any:^(id x) { return [x intValue] == 3; }]);
    test(![(@[@2, @4, @5]) any:^(NSNumber* x) { return [x intValue] == 3; }]);
}
-(void) testMap {
    test([[@[] map:^(id x) { return x; }] isEqualToArray:@[]]);
    test([[(@[@1,@2]) map:^(id x) { return x; }] isEqualToArray:(@[@1,@2])]);
    test([[(@[@1,@2]) map:^(NSNumber* x) { return @([x intValue] + 1); }] isEqualToArray:(@[@2,@3])]);
}
-(void) testFilter {
    test([[@[] filter:^(id x) { return true; }] isEqualToArray:@[]]);
    test([[(@[@1,@2]) filter:^(NSNumber* x) { return true; }] isEqualToArray:(@[@1,@2])]);
    test([[(@[@1,@2]) filter:^(NSNumber* x) { return false; }] isEqualToArray:(@[])]);
    test([[(@[@1,@2]) filter:^(NSNumber* x) { return [x intValue] == 1; }] isEqualToArray:(@[@1])]);
    test([[(@[@1,@2]) filter:^(NSNumber* x) { return [x intValue] == 2; }] isEqualToArray:(@[@2])]);
}
-(void) testSum {
    test([(@[]) sumDouble] == 0);
    test([(@[]) sumNSUInteger] == 0);
    test([(@[]) sumNSInteger] == 0);

    test([(@[@1]) sumDouble] == 1);
    test([(@[@2]) sumNSUInteger] == 2);
    test([(@[@3]) sumNSInteger] == 3);
    
    test([(@[@1.5, @2.75]) sumDouble] == 4.25);
    test([(@[@1, @3]) sumNSUInteger] == 4);
    test([(@[@-1, @4]) sumNSInteger] == 3);
}
-(void) testKeyedBy {
	test([[@[] keyedBy:^id(id value) { return @true; }] isEqual:@{}]);
	test([[@[@1] keyedBy:^id(id value) { return @true; }] isEqual:@{@true : @1}]);
	testThrows(([(@[@1, @2]) keyedBy:^id(id value) { return @true; }]));
	test([[(@[@1, @2]) keyedBy:^id(id value) { return value; }] isEqual:(@{@1 : @1, @2 : @2})]);
	testThrows([(@[@1, @1, @2, @3, @5]) keyedBy:^id(id value) { return @([value intValue]/2); }]);
	test([[(@[@3, @5, @7, @11, @13]) keyedBy:^id(id value) { return @([value intValue]/2); }] isEqual:(@{@1 : @3, @2 : @5, @3 : @7, @5 : @11, @6 : @13})]);
}
-(void) testGroupBy {
	test([[@[] groupBy:^id(id value) { return @true; }] isEqual:@{}]);
	test([[@[@1] groupBy:^id(id value) { return @true; }] isEqual:@{@true : @[@1]}]);
	test([[(@[@1, @2]) groupBy:^id(id value) { return @true; }] isEqual:@{@true : (@[@1, @2])}]);
	test([[(@[@1, @2]) groupBy:^id(id value) { return value; }] isEqual:(@{@1 : @[@1], @2 : @[@2]})]);
	test([[(@[@1, @1, @2, @3, @5]) groupBy:^id(id value) { return @([value intValue]/2); }] isEqual:(@{@0 : @[@1, @1], @1 : @[@2, @3], @2 : @[@5]})]);
}
@end