test_bninput.py
506 Bytes
import unittest
from bninput import *
class Test(unittest.TestCase):
def test_valid_file(self):
result = summary_dag('asia2.bif')
self.assertEqual(result['dag_file'], 'asia2.bif')
self.assertIsInstance(result['nodes'], list)
self.assertIsInstance(result['count_parents'], int)
self.assertIsInstance(result['count_children'], int)
self.assertIsInstance(result['count_in_edges'], int)
self.assertIsInstance(result['count_out_edges'], int)