update unit tests after refactoring

test/inherit/subdir/CustomBudget/__init__.py

-from gwinc.ifo.noises import *
 from gwinc.ifo import PLOT_STYLE
+from gwinc import noise
+from gwinc import nb
+from gwinc.ifo.noises import Strain
 
 
-class QuantumVacuum(nb.Budget):
+class Quantum(nb.Budget):
     """Quantum Vacuum
 
     """
@@ -12,13 +14,13 @@ class QuantumVacuum(nb.Budget):
     )
 
     noises = [
-        QuantumVacuumAS,
-        QuantumVacuumArm,
-        QuantumVacuumSEC,
-        QuantumVacuumFilterCavity,
-        QuantumVacuumInjection,
-        QuantumVacuumReadout,
-        QuantumVacuumQuadraturePhase,
+        noise.quantum.AS,
+        noise.quantum.Arm,
+        noise.quantum.SEC,
+        noise.quantum.FilterCavity,
+        noise.quantum.Injection,
+        noise.quantum.Readout,
+        noise.quantum.QuadraturePhase,
     ]
 
 
@@ -27,8 +29,8 @@ class CustomBudget(nb.Budget):
     name = 'A custom budget'
 
     noises = [
-        QuantumVacuum,
-        Seismic,
+        Quantum,
+        noise.seismic.Seismic,
     ]
 
     calibrations = [

test/inherit/test_inherit.py

@@ -58,7 +58,7 @@ def test_inherit_custom_budget(fpath_join):
     noises = [noise.__name__ for noise in B_inherit.noises]
 
     assert B_inherit.name == 'A custom budget'
-    assert sorted(noises) == ['QuantumVacuum', 'Seismic']
+    assert sorted(noises) == ['Quantum', 'Seismic']
     assert(
         sorted(B_inherit.ifo.diff(B_orig.ifo))
         == sorted([