test_ds.py

import pytest
from ds.dynamic_array import DynamicArray
from ds.stack import Stack
from ds.queue import RingBufferQueue
from ds.hashset import HashSet

def test_dynamic_array():
    arr = DynamicArray()
    arr.append(1)
    arr.append(2)
    assert arr[0] == 1
    assert len(arr) == 2
    assert arr.pop() == 2

def test_stack():
    stack = Stack()
    stack.push(1)
    stack.push(2)
    assert stack.pop() == 2
    assert stack.peek() == 1
    assert not stack.is_empty()

def test_queue():
    queue = RingBufferQueue()
    queue.enqueue(1)
    queue.enqueue(2)
    assert queue.dequeue() == 1
    assert len(queue) == 1

def test_queue_resize():
    queue = RingBufferQueue()
    for i in range(10):
        queue.enqueue(i)
    for i in range(10):
        assert queue.dequeue() == i

def test_hashset():
    hs = HashSet()
    hs.add(1)
    hs.add(2)
    assert hs.contains(1)
    assert not hs.contains(3)
    assert len(hs) == 2