Add temperature beta to Python

Author: DanShort12

parametric_plasma_source/plasma.py

@@ -23,6 +23,7 @@ def __init__(self,
                  pedistal_radius=0.8,
                  ion_density_peaking_factor=1,
                  ion_temperature_peaking_factor=8.06,
+                 ion_temperature_beta=6.0,
                  shafranov_shift=0.0,
                  number_of_bins=100,
                  plasma_type=1,
@@ -44,6 +45,7 @@ def __init__(self,
         self.pedistal_radius = pedistal_radius  # pedistal major rad
         self.ion_density_peaking_factor = ion_density_peaking_factor
         self.ion_temperature_peaking_factor = ion_temperature_peaking_factor
+        self.ion_temperature_beta = ion_temperature_beta
         self.shafranov_shift = shafranov_shift
         self.number_of_bins = number_of_bins
         self.plasma_type = plasma_type  # 0 = L mode anything else H/A mode
@@ -109,6 +111,17 @@ def ion_temperature_peaking_factor(self, ion_temperature_peaking_factor):
         else:
             self._ion_temperature_peaking_factor = ion_temperature_peaking_factor
 
+    @property
+    def ion_temperature_beta(self):
+        return self._ion_temperature_beta
+
+    @ion_temperature_beta.setter
+    def ion_temperature_beta(self, ion_temperature_beta):
+        if ion_temperature_beta < 0:
+            raise ValueError('ion_temperature_beta is out of range')
+        else:
+            self._ion_temperature_beta = ion_temperature_beta
+
     @property
     def ion_density_peaking_factor(self):
         return self._ion_density_peaking_factor
@@ -329,6 +342,7 @@ def export_plasma_source(self, output_filename):
             "pedistal_radius": self.pedistal_radius,
             "ion_density_peaking_factor": self.ion_density_peaking_factor,
             "ion_temperature_peaking_factor": self.ion_temperature_peaking_factor,
+            "ion_temperature_beta": self.ion_temperature_beta,
             "minor_radius": self.minor_radius,
             "major_radius": self.major_radius,
             "elongation": self.elongation,