Merge pull request #1 from mbonkile/Exp

Exp

Author: RuiheLi

pybamm/__init__.py

@@ -222,6 +222,7 @@
 #
 from .experiments.experiment import Experiment
 from . import experiments
+from .experiments.experiment_steps import CC
 
 #
 # Plotting

pybamm/experiments/experiment_steps.py

@@ -0,0 +1,26 @@
+#
+# Experiment class
+#
+
+import numpy as np
+import re
+
+
+class CC:
+    """
+    Description:
+
+
+    """
+
+    def __init__(
+        self,
+        C_rate="",
+        temperature=None,
+        duration=None,
+        upper_cutoff=None,
+    ):
+        self.c_rate = C_rate
+        self.temperature = temperature
+        self.duration = duration
+        self.upper_cutoff = upper_cutoff

tests/unit/test_experiments/test_experiment_steps.py

@@ -0,0 +1,33 @@
+#
+# Test the base experiment class
+#
+import pybamm
+import numpy as np
+import unittest
+import pandas as pd
+import os
+
+
+class TestExperimentSteps(unittest.TestCase):
+    def test_cc(self):
+        expcc = pybamm.CC(1)
+        self.assertEqual(expcc.c_rate, 1)
+        self.assertEqual(expcc.temperature, None)
+        self.assertEqual(expcc.duration, None)
+        self.assertEqual(expcc.upper_cutoff, None)
+
+        expccall = pybamm.CC(1, 298, 3600, 4.2)
+        self.assertEqual(expccall.c_rate, 1)
+        self.assertEqual(expccall.temperature, 298)
+        self.assertEqual(expccall.duration, 3600)
+        self.assertEqual(expccall.upper_cutoff, 4.2)
+
+
+if __name__ == "__main__":
+    print("Add -v for more debug output")
+    import sys
+
+    if "-v" in sys.argv:
+        debug = True
+    pybamm.settings.debug_mode = True
+    unittest.main()