unit test for CWAVLTree -objectIsInTree

Machx · Aug 15, 2012 · 8543386 · 8543386
1 parent d118a93
commit 8543386
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/Source/CWAVLTreeTests.m b/Source/CWAVLTreeTests.m
@@ -31,4 +31,20 @@ -(void)testBasicInsertion
 	[tree addObject:@10];
 }
 
+-(void)testObjectIsInTree
+{
+	CWAVLTree *tree = [[CWAVLTree alloc] init];
+
+	[tree addObject:@42];
+	[tree addObject:@50];
+	[tree addObject:@8];
+
+	STAssertTrue([tree objectIsInTree:@42], nil);
+	STAssertTrue([tree objectIsInTree:@50], nil);
+	STAssertTrue([tree objectIsInTree:@8], nil);
+	STAssertFalse([tree objectIsInTree:@100], nil);
+	STAssertFalse([tree objectIsInTree:@1], nil);
+	STAssertFalse([tree objectIsInTree:@1000], nil);
+}
+
 @end