#29: Test bracket access functionality

philippkraft · May 9, 2018 · 37eb074 · 37eb074
1 parent a0f9b57
commit 37eb074
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/test/solute_access.py b/test/solute_access.py
@@ -0,0 +1,22 @@
+import cmf
+import unittest
+
+
+class TestSoluteAccess(unittest.TestCase):
+    def testConc(self):
+        p = cmf.project('X Y Z')
+        X, Y, Z = p.solutes
+        ws = p.NewStorage('storage')
+        self.assertEqual(repr(ws[X]), '{storage}[X]')
+        self.assertEqual(ws.conc(X), ws[X].conc)
+
+    def testState(self):
+        p = cmf.project('X Y Z')
+        X, Y, Z = p.solutes
+        ws = p.NewStorage('storage')
+        ws.volume = 1.0
+        ws[X].state = 1.0
+        ws.volume = 2.0
+        self.assertAlmostEquals(ws[X].conc, 0.5)
+        self.assertAlmostEquals(ws[X].state, 1.0)
+        self.assertAlmostEquals(ws[Y].conc, 0.0)