avl_test.cpp

#include "../src/avl.h"

/* Custom data-type for testing purposes */
struct customNode
{
    /* An integer */
    int val;

    /* A character */
    char c;

    /* Provide definition of < to help compare 2 'CustomNode's */
    bool operator<( const struct customNode& other )
    {
        if( val < other.val )
        {
            return true;
        }
        if( val == other.val )
        {
            return c < other.c;
        }
        return false;
    }

    /* Provide definition of == to help compare 2 'CustomNode' for equality */
    bool operator==( const struct customNode& other )
    {
        return ( val == other.val && c == other.c );
    }

    /* Provide definition of > to help compare 2 'CustomNode's */
    bool operator>( const struct customNode& other )
    {
        return !( *this < other ) && !( *this == other );
    }

    /* A friendly function to output CustomNode easily */
    friend std::ostream& operator<<( std::ostream& out, struct customNode& obj )
    {
        out << obj.val << " " << obj.c;
        return out;
    }
};
/* A handy alias for struct customNode */
typedef struct customNode CustomNode;

int main( void )
{
    /* Code for testing each public API */

    /* First, lets' create an AVL tree of integers from 11-20 */
    AVL::Tree< int > t1;
    for( int i = 11; i <= 20; ++i )
    {
        assert( t1.insert( i ) );
    }

    /* Size of t1 should be 10 */
    assert( t1.size() == 10 );

    /* 15 should be present and 25 should not be */
    assert( t1.search( 15 ) );
    assert( !t1.search( 25 ) );

    /* Let's do a successful delete of 15 and an unsuccessful delete of 30 */
    assert( t1.findAndDeleteByCopying( 15 ) );
    assert( !t1.findAndDeleteByCopying( 30 ) );

    /* Let's do a forward inorder iteration */
    AVL::Iterator< int > fwd( t1.begin() );
    /* Iterate only if 'ptr' is not null
       'ptr' points to the 'Node< int >' object and 'el' is the required
       value */
    while( fwd.ptr )
    {
        /* Print element value */
        std::cout << fwd.ptr->el << "\n";
        /* Fetch next */
        t1.nextNode( fwd );
    }
    std::cout << "\n";

    /* Let's do a backward inorder iteration */
    AVL::Iterator< int > bwd( t1.rbegin() );
    /* Iterate only if 'ptr' is not null */
    while( bwd.ptr )
    {
        /* Print element value */
        std::cout << bwd.ptr->el << "\n";
        /* Fetch previous */
        t1.prevNode( bwd );
    }
    std::cout << "\n";

    /* Let's use getIndexGivenValue */
    /* 17 is the 6th element */
    assert( t1.getIndexGivenValue( 17 ) == 6 );
    /* 25 doesn't exist */
    assert( t1.getIndexGivenValue( 25 ) == 0 );

    /* Let's use getValueGivenIndex */
    /* 16 is at index 5 */
    assert( t1.getValueGivenIndex( 5 )->el == 16 );
    /* t1.size() + 1 index is invalid */
    assert( t1.getValueGivenIndex( t1.size() + 1 ) == 0 );

    /* Let's use lowerbound */
    /* lowerbound of 14 is 14 */
    assert( t1.lowerbound( 14 )->el == 14 );
    /* lowerbound of 15 is 16 */
    assert( t1.lowerbound( 15 )->el == 16 );
    /* lowerbound of 9 is 11 */
    assert( t1.lowerbound( 9 )->el == 11 );
    /* lowerbound of 21 doesn't exist in t1 */
    assert( t1.lowerbound( 21 ) == 0 );

    /* Create t2 and t3 using copy constructor and assignment operator
       respectively */
    AVL::Tree< int > t2( t1 ), t3 = t2;

    /* Insert 50 in t2 and output it */
    t2.insert( 50 );
    /* Use previously defined iterator */
    fwd.set( t2.begin() );
    while( fwd.ptr )
    {
        /* Print element value */
        std::cout << fwd.ptr->el << "\n";
        /* Fetch next */
        t2.nextNode( fwd );
    }
    std::cout << "\n";

    /* Clear t3 and output it */
    t3.clear();
    fwd.set( t3.begin() );
    while( fwd.ptr )
    {
        /* Print element value */
        std::cout << fwd.ptr->el << "\n";
        t3.nextNode( fwd );
    }
    std::cout << "\n";

    /* Let's try using some custom data-type */
    AVL::Tree< CustomNode > t4;
    for( int i = 10; i <= 20; ++i )
    {
        assert( t4.insert( { i, 'a' } ) );
    }
    for( int i = 21; i <= 30; ++i )
    {
        assert( t4.insert( { i, 'b' } ) );
    }

    /* Size of t4 should be 21 */
    assert( t4.size() == 21 );

    /* Let's output it */
    /* Forward iteration */
    AVL::Iterator< CustomNode > it( t4.begin() );
    while( it.ptr )
    {
        /* Print element value */
        std::cout << it.ptr->el << "\n";
        t4.nextNode( it );
    }
    std::cout << "\n";
    /* Backward iteration */
    it.set( t4.rbegin() );
    while( it.ptr )
    {
        /* Print element value */
        std::cout << it.ptr->el << "\n";
        t4.prevNode( it );
    }

    /* Let's do a heavy testing using 1 million integers */
    AVL::Tree< int > t5;
    for( int i = 1; i <= 1e6; ++i )
    {
        assert( t5.insert( i ) );
    }

    /* Iterate over t5 and test whether the ith value is equal to i */
    int i = 1;
    fwd.set( t5.begin() );
    while( fwd.ptr )
    {
        assert( fwd.ptr->el == i );
        ++i;
        t5.nextNode( fwd );
    }

    /* Similarly, iterate backward and test whether the ith value is
       equal to i */
    --i;
    bwd.set( t5.rbegin() );
    while( bwd.ptr )
    {
        assert( bwd.ptr->el == i );
        --i;
        t5.prevNode( bwd );
    }
    /* Testing complete */
    return 0;
}