public release 2.2.0 - see README.md and CHANGES.md for details
This commit is contained in:
@ -32,11 +32,11 @@ import pmsco.project as mp
|
||||
class TestPopulation(unittest.TestCase):
|
||||
def setUp(self):
|
||||
random.seed(0)
|
||||
self.domain = mp.Domain()
|
||||
self.model_space = mp.ModelSpace()
|
||||
|
||||
self.domain.add_param('A', 1.5, 1.0, 2.0, 0.2)
|
||||
self.domain.add_param('B', 2.5, 2.0, 3.0, 0.25)
|
||||
self.domain.add_param('C', 3.5, 3.5, 3.5, 0.0)
|
||||
self.model_space.add_param('A', 1.5, 1.0, 2.0, 0.2)
|
||||
self.model_space.add_param('B', 2.5, 2.0, 3.0, 0.25)
|
||||
self.model_space.add_param('C', 3.5, 3.5, 3.5, 0.0)
|
||||
self.expected_popsize = 30
|
||||
self.expected_names = ('_model', '_rfac', 'A', 'B', 'C')
|
||||
|
||||
@ -57,7 +57,7 @@ class TestPopulation(unittest.TestCase):
|
||||
pass
|
||||
|
||||
def test_setup(self):
|
||||
self.pop.setup(self.domain)
|
||||
self.pop.setup(self.model_space)
|
||||
self.assertEqual(self.pop.positions.dtype.names, self.expected_names)
|
||||
self.assertEqual(self.pop.positions.shape, (self.expected_popsize,))
|
||||
self.assertEqual(self.pop.model_count, self.expected_popsize)
|
||||
|
Reference in New Issue
Block a user