# coding: utf-8
"""
Braginskii coefficients
=========================

A short example of how to calculate classical transport coefficients
from Bragiński's theory.
"""

from astropy import units as u
from plasmapy import physics
import matplotlib.pyplot as plt
import numpy as np

#####################################################
# We'll use some sample ITER data, without much regard for whether
# the regime is even fit for classical transport theory:

thermal_energy_per_electron = 8.8 * u.keV
electron_concentration = 10.1e19 / u.m**3

thermal_energy_per_ion = 8.0 * u.keV
ion_concentration = electron_concentration
ion_particle = 'D+'  # a crude approximation

######################################################
# We now make the default classical_transport object:

braginskii = physics.transport.classical_transport(thermal_energy_per_electron,
                                                   electron_concentration,
                                                   thermal_energy_per_ion,
                                                   ion_concentration,
                                                   ion_particle)

######################################################
# These variables are calculated during initialization and can be
# referred to straight away:

print(braginskii.coulomb_log_ei)
print(braginskii.coulomb_log_ii)
print(braginskii.hall_e)
print(braginskii.hall_i)

######################################################
# These quantities are not calculated during initialization and can be
# referred to via methods. To signify the need to calculate them, we
# call them via (). They could be made to act like variables via @Property,
# but I'm not sure that's wise.

print(braginskii.resistivity())
print(braginskii.thermoelectric_conductivity())

@@ -211,23 +211,64 @@
 mapy
- import physics
+.physics.transport.braginskii import classical_transport
 %0Aimp
@@ -829,34 +829,16 @@
 nskii = 
-physics.transport.
 classica
@@ -911,34 +911,16 @@
         
-                  
 electron
@@ -968,34 +968,16 @@
         
-         
-         
 thermal_
@@ -1025,34 +1025,16 @@
         
-              
-    
 ion_conc
@@ -1044,34 +1044,16 @@
 ration,%0A
-                  
         

Fix import path in example