test.py

import networkx as nx
from topology_shape_metrics.TSM import TSM
import unittest
from matplotlib import pyplot as plt

class TestGML(unittest.TestCase):
    def _test(gml_filename):
        G = nx.Graph(nx.read_gml(gml_filename))
        pos = {node: eval(node) for node in G}

        tsm = TSM(G, pos)
        tsm.display()
        plt.savefig(gml_filename.replace(
            "inputs", "outputs").replace(".gml", ".nolp.svg"))

        tsm = TSM(G, pos, checkit=False, uselp=True)
        tsm.display()
        plt.savefig(gml_filename.replace(
            "inputs", "outputs").replace(".gml", ".lp.svg"))

    def test_01(self):
        TestGML._test("test/inputs/case1.gml")

    def test_02(self):
        TestGML._test("test/inputs/case2.gml")

    def test_03(self):
        TestGML._test("test/inputs/case3.gml")

    def test_04(self):
        TestGML._test("test/inputs/case4.gml")


class TestGrid(unittest.TestCase):
    def _test_grid(i, j):
        G = nx.grid_2d_graph(i, j)
        pos = {node: node for node in G}
        TSM(G, pos)
        TSM(G)

    def test_01(self):
        TestGrid._test_grid(1, 2)

    def test_02(self):
        TestGrid._test_grid(2, 1)

    def test_03(self):
        TestGrid._test_grid(1, 5)

    def test_04(self):
        TestGrid._test_grid(5, 1)

    def test_05(self):
        TestGrid._test_grid(5, 5)

    def test_06(self):
        TestGrid._test_grid(3, 3)

    def test_07(self):
        TestGrid._test_grid(9, 9)


if __name__ == '__main__':
    res = unittest.main(verbosity=3, exit=False)