Source code for HelloGraph.test

import unittest
from HelloGraph.challenge import HelloGraphChallenge

[docs]class HelloGraphTest(unittest.TestCase):

[docs]    def setUp(self):
        self.challenge = HelloGraphChallenge()

[docs]    def test__init__(self):
        self.assertIsInstance(self.challenge, HelloGraphChallenge)
        self.assertIn('5->6', self.challenge.sample)
        self.assertIn('2->3->4', self.challenge.expect)

[docs]    def test_build(self):
        self.challenge.read()
        self.challenge.build()
        self.assertEqual(2, self.challenge.model.start)
        self.assertEqual(4, self.challenge.model.stop)
        self.assertIn(4, self.challenge.model.edges[3])
        self.assertEqual(10, self.challenge.model.weights[(3, 4)])

[docs]    def test_format(self):
        self.challenge.result.weight = 14
        self.challenge.result.path = [2, 3, 4]
        self.challenge.format()
        self.assertEqual(self.challenge.expectation(), self.challenge.output)

[docs]    def test_full_integration(self):
        self.challenge.main()
        self.assertEqual(self.challenge.expectation(), self.challenge.output)