test_graph_id_generator.py

"""
B-ASIC test suite for graph id generator.
"""

from b_asic.graph_id import GraphIDGenerator, GraphID

import pytest

def test_empty_string_generator():
    """Test the graph id generator for an empty string type."""
    graph_id_generator = GraphIDGenerator()
    assert graph_id_generator.get_next_id("") == "1"
    assert graph_id_generator.get_next_id("") == "2"


def test_normal_string_generator():
    """"Test the graph id generator for a normal string type."""
    graph_id_generator = GraphIDGenerator()
    assert graph_id_generator.get_next_id("add") == "add1"
    assert graph_id_generator.get_next_id("add") == "add2"

def test_different_strings_generator():
    """Test the graph id generator for different strings."""
    graph_id_generator = GraphIDGenerator()
    assert graph_id_generator.get_next_id("sub") == "sub1"
    assert graph_id_generator.get_next_id("mul") == "mul1"
    assert graph_id_generator.get_next_id("sub") == "sub2"
    assert graph_id_generator.get_next_id("mul") == "mul2"