from django.test import TestCase

from epdb.logic import SNode, SEdge, SPathway


class SObjectTest(TestCase):

    def setUp(self):
        pass

    def test_snode_eq(self):
        snode1 = SNode('CN1C2C(N(C(N(C)C=2N=C1)=O)C)=O', 0)
        snode2 = SNode('CN1C2C(N(C(N(C)C=2N=C1)=O)C)=O', 0)
        assert snode1 == snode2

    def test_snode_hash(self):
        pass

    def test_sedge_eq(self):
        sedge1 = SEdge([SNode('CN1C2C(N(C(N(C)C=2N=C1)=O)C)=O', 0)],
                       [SNode('CN1C(=O)NC2=C(C1=O)N(C)C=N2', 1), SNode('C=O', 1)],
                       rule=None)
        sedge2 = SEdge([SNode('CN1C2C(N(C(N(C)C=2N=C1)=O)C)=O', 0)],
                       [SNode('CN1C(=O)NC2=C(C1=O)N(C)C=N2', 1), SNode('C=O', 1)],
                       rule=None)
        assert sedge1 == sedge2

    def test_sedge_hash(self):
        pass

    def test_spathway(self):
        pw = SPathway()