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(graph_id_generator):
    """Test the graph id generator for an empty string type."""
    assert graph_id_generator.get_next_id("") == GraphID("", 1)
    assert graph_id_generator.get_next_id("") == GraphID("", 2)


def test_normal_string_generator(graph_id_generator):
    """"Test the graph id generator for a normal string type."""
    assert graph_id_generator.get_next_id("add") == GraphID("add", 1)
    assert graph_id_generator.get_next_id("add") == GraphID("add", 2)

def test_different_strings_generator(graph_id_generator):
    """Test the graph id generator for different strings."""
    assert graph_id_generator.get_next_id("sub") == GraphID("sub", 1)
    assert graph_id_generator.get_next_id("mul") == GraphID("mul", 1)
    assert graph_id_generator.get_next_id("sub") == GraphID("sub", 2)
    assert graph_id_generator.get_next_id("mul") == GraphID("mul", 2)