def test_constructor_FDataGrid_setting(self):
data = self.dummy_data
a = FDataGrid(data)
np.testing.assert_equal(a.extrapolation, None)
a = FDataGrid(data, extrapolation=PeriodicExtrapolation())
np.testing.assert_equal(a.extrapolation, PeriodicExtrapolation())
a = FDataGrid(data, extrapolation=BoundaryExtrapolation())
np.testing.assert_equal(a.extrapolation, BoundaryExtrapolation())
np.testing.assert_equal(a.extrapolation == ExceptionExtrapolation(),
False)
a = FDataGrid(data, extrapolation=ExceptionExtrapolation())
np.testing.assert_equal(a.extrapolation, ExceptionExtrapolation())
a = FDataGrid(data, extrapolation=FillExtrapolation(0))
np.testing.assert_equal(a.extrapolation, FillExtrapolation(0))
np.testing.assert_equal(a.extrapolation == FillExtrapolation(1), False)
def test_constructor_FDataBasis_setting(self):
coeff = self.dummy_data
basis = Fourier(n_basis=3)
a = FDataBasis(basis, coeff)
np.testing.assert_equal(a.extrapolation, None)
a = FDataBasis(basis, coeff, extrapolation=PeriodicExtrapolation())
np.testing.assert_equal(a.extrapolation, PeriodicExtrapolation())
np.testing.assert_equal(a.extrapolation == BoundaryExtrapolation(),
False)
a = FDataBasis(basis, coeff, extrapolation=BoundaryExtrapolation())
np.testing.assert_equal(a.extrapolation, BoundaryExtrapolation())
a = FDataBasis(basis, coeff, extrapolation=ExceptionExtrapolation())
np.testing.assert_equal(a.extrapolation, ExceptionExtrapolation())
a = FDataBasis(basis, coeff, extrapolation=FillExtrapolation(0))
np.testing.assert_equal(a.extrapolation, FillExtrapolation(0))
np.testing.assert_equal(a.extrapolation == FillExtrapolation(1), False)
def test_FDataGrid_setting(self):
data = self.dummy_data
a = FDataGrid(data)
a.extrapolation = PeriodicExtrapolation()
np.testing.assert_equal(a.extrapolation, PeriodicExtrapolation())
a.extrapolation = "bounds"
np.testing.assert_equal(a.extrapolation, BoundaryExtrapolation())
a.extrapolation = "exception"
np.testing.assert_equal(a.extrapolation, ExceptionExtrapolation())
a.extrapolation = "zeros"
np.testing.assert_equal(a.extrapolation, FillExtrapolation(0))
np.testing.assert_equal(a.extrapolation == FillExtrapolation(1), False)
def test_FDataBasis_setting(self):
coeff = self.dummy_data
basis = Fourier(n_basis=3)
a = FDataBasis(basis, coeff)
a.extrapolation = "periodic"
np.testing.assert_equal(a.extrapolation, PeriodicExtrapolation())
a.extrapolation = "bounds"
np.testing.assert_equal(a.extrapolation, BoundaryExtrapolation())
a.extrapolation = "exception"
np.testing.assert_equal(a.extrapolation, ExceptionExtrapolation())
a.extrapolation = "zeros"
np.testing.assert_equal(a.extrapolation, FillExtrapolation(0))
a.extrapolation = "nan"
np.testing.assert_equal(a.extrapolation, FillExtrapolation(np.nan))