Skip to content

Commit

Permalink
#191 more dict reordering bugs
Browse files Browse the repository at this point in the history
  • Loading branch information
martinjrobins committed May 9, 2019
1 parent 99a2e1a commit 151930c
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions tests/unit/test_discretisations/test_discretisation.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ def test_concatenate_in_order(self):
disc._y_slices = {c.id: slice(0, 1), a.id: slice(2, 3), b.id: slice(3, 4)}
result = disc._concatenate_in_order(initial_conditions)

self.assertIsInstance(result, pybamm.NumpyConcatenation, check_complete=True)
self.assertIsInstance(result, pybamm.NumpyConcatenation)
self.assertEqual(result.children[0].evaluate(), 1)
self.assertEqual(result.children[1].evaluate(), 2)
self.assertEqual(result.children[2].evaluate(), 3)
Expand Down Expand Up @@ -397,11 +397,13 @@ def test_process_model_ode(self):
disc.process_model(model)
y0 = model.concatenated_initial_conditions
y0_expect = np.array([])
for var, init in model.initial_conditions.items():
if var.id == c.id:
vect = init.evaluate() * np.ones_like(combined_submesh[0].nodes)
for var_id, _ in sorted(disc._y_slices.items(), key=lambda kv: kv[1]):
if var_id == c.id:
vect = 2 * np.ones_like(combined_submesh[0].nodes)
elif var_id == T.id:
vect = 5 * np.ones_like(mesh["negative electrode"][0].nodes)
else:
vect = init.evaluate() * np.ones_like(mesh[var.domain[0]][0].nodes)
vect = 8 * np.ones_like(mesh["negative electrode"][0].nodes)

y0_expect = np.concatenate([y0_expect, vect])

Expand Down

0 comments on commit 151930c

Please sign in to comment.