class test_OpacityTable(object): """OpacityTable""" def setup(self): """Set up this test suite""" self.o = OpacityTable("GN93hz",load=False) def test_solar_composition(self): """Interpolated Values Succeed. logrho=[0.3,-4.0], logT=[6.3,5.0]""" self.o.composition(X=0.7,Y=0.28) assert self.o.n == 72, u"Table Mismatch, %g ≠ %g" % (self.o.n,72) v1 = self.o.lookup(logrho=0.3,logT=6.3) a1 = 1.885 assert v1 - a1 < 1e-10, u"κ mismatch: %g ≠ %g" % (v1, a1) v2 = self.o.lookup(logrho=-4.0,logT=5.0) a2 = 3.436 assert v2 - a2 < 1e-10, u"κ mismatch: %g ≠ %g" % (v2, a2) @nt.raises(AssertionError) def test_sanity_comp(self): """Composition should fail if X+Y > 1.0.""" self.o.composition(X=0.7,Y=0.7) @nt.raises(ValueError) def test_lower_logR_bound(self): """Values below logR=-8.0 fail.""" self.o.composition(X=0.70,Y=0.28) logrho, logT = self.o.invert_points(logR=-9.0,logT=5.00).T v1 = self.o.lookup(logrho=logrho,logT=logT) @nt.raises(ValueError) def test_upper_logR_bound(self): """Values above logR=1.0 fail.""" self.o.composition(X=0.70,Y=0.28) logrho, logT = self.o.invert_points(logR=2.0,logT=5.00).T v1 = self.o.lookup(logrho=logrho,logT=logT) @nt.raises(ValueError) def test_lower_logT_bound(self): """Values below logT=3.00 fail.""" self.o.composition(X=0.70,Y=0.28) logrho, logT = self.o.invert_points(logR=-4.0,logT=3.00).T v1 = self.o.lookup(logrho=logrho,logT=logT) @nt.raises(ValueError) def test_upper_logT_bound(self): """Values above logT=9.00 fail.""" self.o.composition(X=0.70,Y=0.28) logrho, logT = self.o.invert_points(logR=-4.0,logT=9.00).T v1 = self.o.lookup(logrho=logrho,logT=logT) @nt.raises(ValueError) def test_corner_a_NaNs(self): """Values in the bottom right corner of the table should fail. logR=0.5, logT=8.5""" self.o.composition(X=0.70,Y=0.28) logrho, logT = self.o.invert_points(logR=0.5,logT=8.50).T v1 = self.o.lookup(logrho=logrho,logT=logT) assert v1 == 1.0 @nt.raises(ValueError) def test_corner_b_NaNs(self): """Values in the top left corner of the Y=1.000 table should fail. logR=-8.0, logT=3.75""" self.o.composition(X=0.00,Y=1.00) logrho, logT = self.o.invert_points(logR=-8.0,logT=3.75).T v1 = self.o.lookup(logrho=logrho,logT=logT) assert v1 == 1.0 def test_corner_a_valid(self): """Values in the bottom left corner of the table should succeed. logR=-8.0, logT=8.70""" self.o.composition(X=0.70,Y=0.28) logrho, logT = self.o.invert_points(logR=-8.0,logT=8.70).T v1 = self.o.lookup(logrho=logrho,logT=logT) a1 = -0.582 assert v1 - a1 < 1e-10, u"κ mismatch: %g ≠ %g" % (v1, a1) def test_corner_b_valid(self): """Values in the top right corner of the table should succeed. logR=1.0, logT=3.75""" self.o.composition(X=0.70,Y=0.28) logrho, logT = self.o.invert_points(logR=1.0,logT=3.75).T v1 = self.o.lookup(logrho=logrho,logT=logT) a1 = 0.131 assert v1 - a1 < 1e-10, u"κ mismatch: %g ≠ %g" % (v1, a1) def test_midtable_valid(self): """Direct values succeed. logR=-4.0, logT=5.45""" self.o.composition(X=0.70,Y=0.28) logrho, logT = self.o.invert_points(logR=-4.0,logT=5.45).T v1 = self.o.lookup(logrho=logrho,logT=logT) a1 = 0.680 assert v1 - a1 < 1e-10, u"κ mismatch: %g ≠ %g" % (v1, a1)
for i,(logT,logrho) in enumerate(zip(logTs,logrhos)): kappa = opacity1.lookup(logrho=logrho,logT=logT) print u"1: log(T)=%5.3f, log(ρ)=%6.3f → κ=%5.3f" % (logT,logrho,kappa) kappa = opacity2.lookup(logrho=logrho,logT=logT) print u"2: log(T)=%5.3f, log(ρ)=%6.3f → κ=%5.3f" % (logT,logrho,kappa) kappa = opacity3.lookup(logrho=logrhosmall[i],logT=logTsmall[i]) print u"3: log(T)=%5.3f, log(ρ)=%6.3f → κ=%5.3f" % (logTsmall[i],logrhosmall[i],kappa) kappa = opacity2.lookup(logrho=logrhosmall[i],logT=logTsmall[i]) print u"2s log(T)=%5.3f, log(ρ)=%6.3f → κ=%5.3f" % (logTsmall[i],logrhosmall[i],kappa) opacity1.composition(X=0.00,Y=1.00) opacity2.composition(X=0.00,Y=1.00) print u"1: Fixed composition at X=%.3g,Y=%.3g,Z=%.3g" % (opacity1.X,opacity1.Y,opacity1.Z) print u"2: Fixed composition at X=%.3g,Y=%.3g,Z=%.3g" % (opacity2.X,opacity2.Y,opacity2.Z) print u"1: Using table %d" % opacity1.n print u"2: Using table %d" % opacity2.n logrho, logT = opacity1.invert_points(logR=-8.0,logT=3.75).T try: kappa = opacity1.lookup(logrho=logrho,logT=logT) print u"1: log(T)=%5.3f, log(ρ)=%6.3f → κ=%5.3f" % (logT,logrho,kappa) except ValueError: print u"1: log(T)=%5.3f, log(ρ)=%6.3f → κ=NaN" % (logT,logrho) try: kappa = opacity2.lookup(logrho=logrho,logT=logT) print u"2: log(T)=%5.3f, log(ρ)=%6.3f → κ=%5.3f" % (logT,logrho,kappa) except ValueError: print u"2: log(T)=%5.3f, log(ρ)=%6.3f → κ=NaN" % (logT,logrho)
class test_OpacityTable(object): """OpacityTable""" def setup(self): """Set up this test suite""" self.o = OpacityTable("GN93hz", load=False) def test_solar_composition(self): """Interpolated Values Succeed. logrho=[0.3,-4.0], logT=[6.3,5.0]""" self.o.composition(X=0.7, Y=0.28) assert self.o.n == 72, u"Table Mismatch, %g ≠ %g" % (self.o.n, 72) v1 = self.o.lookup(logrho=0.3, logT=6.3) a1 = 1.885 assert v1 - a1 < 1e-10, u"κ mismatch: %g ≠ %g" % (v1, a1) v2 = self.o.lookup(logrho=-4.0, logT=5.0) a2 = 3.436 assert v2 - a2 < 1e-10, u"κ mismatch: %g ≠ %g" % (v2, a2) @nt.raises(AssertionError) def test_sanity_comp(self): """Composition should fail if X+Y > 1.0.""" self.o.composition(X=0.7, Y=0.7) @nt.raises(ValueError) def test_lower_logR_bound(self): """Values below logR=-8.0 fail.""" self.o.composition(X=0.70, Y=0.28) logrho, logT = self.o.invert_points(logR=-9.0, logT=5.00).T v1 = self.o.lookup(logrho=logrho, logT=logT) @nt.raises(ValueError) def test_upper_logR_bound(self): """Values above logR=1.0 fail.""" self.o.composition(X=0.70, Y=0.28) logrho, logT = self.o.invert_points(logR=2.0, logT=5.00).T v1 = self.o.lookup(logrho=logrho, logT=logT) @nt.raises(ValueError) def test_lower_logT_bound(self): """Values below logT=3.00 fail.""" self.o.composition(X=0.70, Y=0.28) logrho, logT = self.o.invert_points(logR=-4.0, logT=3.00).T v1 = self.o.lookup(logrho=logrho, logT=logT) @nt.raises(ValueError) def test_upper_logT_bound(self): """Values above logT=9.00 fail.""" self.o.composition(X=0.70, Y=0.28) logrho, logT = self.o.invert_points(logR=-4.0, logT=9.00).T v1 = self.o.lookup(logrho=logrho, logT=logT) @nt.raises(ValueError) def test_corner_a_NaNs(self): """Values in the bottom right corner of the table should fail. logR=0.5, logT=8.5""" self.o.composition(X=0.70, Y=0.28) logrho, logT = self.o.invert_points(logR=0.5, logT=8.50).T v1 = self.o.lookup(logrho=logrho, logT=logT) assert v1 == 1.0 @nt.raises(ValueError) def test_corner_b_NaNs(self): """Values in the top left corner of the Y=1.000 table should fail. logR=-8.0, logT=3.75""" self.o.composition(X=0.00, Y=1.00) logrho, logT = self.o.invert_points(logR=-8.0, logT=3.75).T v1 = self.o.lookup(logrho=logrho, logT=logT) assert v1 == 1.0 def test_corner_a_valid(self): """Values in the bottom left corner of the table should succeed. logR=-8.0, logT=8.70""" self.o.composition(X=0.70, Y=0.28) logrho, logT = self.o.invert_points(logR=-8.0, logT=8.70).T v1 = self.o.lookup(logrho=logrho, logT=logT) a1 = -0.582 assert v1 - a1 < 1e-10, u"κ mismatch: %g ≠ %g" % (v1, a1) def test_corner_b_valid(self): """Values in the top right corner of the table should succeed. logR=1.0, logT=3.75""" self.o.composition(X=0.70, Y=0.28) logrho, logT = self.o.invert_points(logR=1.0, logT=3.75).T v1 = self.o.lookup(logrho=logrho, logT=logT) a1 = 0.131 assert v1 - a1 < 1e-10, u"κ mismatch: %g ≠ %g" % (v1, a1) def test_midtable_valid(self): """Direct values succeed. logR=-4.0, logT=5.45""" self.o.composition(X=0.70, Y=0.28) logrho, logT = self.o.invert_points(logR=-4.0, logT=5.45).T v1 = self.o.lookup(logrho=logrho, logT=logT) a1 = 0.680 assert v1 - a1 < 1e-10, u"κ mismatch: %g ≠ %g" % (v1, a1)