32 lines
950 B
Python
32 lines
950 B
Python
from diceplayer.shared.environment.atom import Atom
|
|
from diceplayer.shared.environment.molecule import Molecule
|
|
from diceplayer.shared.environment.system import System
|
|
|
|
import unittest
|
|
|
|
|
|
class TestSystem(unittest.TestCase):
|
|
def test_class_instantiation(self):
|
|
system = System()
|
|
|
|
self.assertIsInstance(system, System)
|
|
|
|
def test_add_type(self):
|
|
system = System()
|
|
system.add_type(0, Molecule('test'))
|
|
|
|
self.assertIsInstance(system.molecule, list)
|
|
self.assertIsInstance(system.nmols, list)
|
|
|
|
with self.assertRaises(TypeError) as ex:
|
|
system.add_type(0, 'test')
|
|
self.assertEqual(ex.exception, 'Error: molecule is not a Molecule instance')
|
|
|
|
with self.assertRaises(TypeError) as ex:
|
|
system.add_type('test', Molecule('test'))
|
|
self.assertEqual(ex.exception, 'Error: nmols is not an integer')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|