コード例 #1
0
def test_signal_size():
    spectrum = np.array([5., 4., 3., 2., 2., 1., 1., 1., 0., 0.])
    ri = ReducedIntensity1D(spectrum)
    pdfgen = PDFGenerator1D(ri)
    pdf = pdfgen.get_pdf(s_min=0, s_max=10)
    assert isinstance(pdf, PairDistributionFunction1D)

    ri = ReducedIntensity1D([spectrum])
    pdfgen = PDFGenerator1D(ri)
    pdf = pdfgen.get_pdf(s_min=0,s_max=10)
    assert isinstance(pdf, PairDistributionFunction1D)

    ri = ReducedIntensity1D([[spectrum]])
    pdfgen = PDFGenerator1D(ri)
    pdf = pdfgen.get_pdf(s_min=0, s_max=10)
    assert isinstance(pdf, PairDistributionFunction1D)

    ri = ReducedIntensity1D([[[spectrum]]])
    pdfgen = PDFGenerator1D(ri)
    pdf = pdfgen.get_pdf(s_min=0, s_max=10)
    shape = pdf.data.shape
    assert shape == (1,1,1,2000)

    ri = ReducedIntensity1D([[[spectrum]]])
    pdfgen = PDFGenerator1D(ri)
    pdf = pdfgen.get_pdf(s_min=0, s_max=10, r_min=0, r_max=8)
    shape = pdf.data.shape
    assert shape == (1,1,1,800)
コード例 #2
0
ファイル: test_pdf_generator.py プロジェクト: ptim0626/pyxem
def test_s_limit_failure(reduced_intensity1d):
    pdfgen = PDFGenerator1D(reduced_intensity1d)

    with pytest.raises(
            ValueError,
            match="User specified s_max is larger than the maximum"):
        pdf3 = pdfgen.get_pdf(s_min=0, s_max=15)
コード例 #3
0
def test_s_limit_failure(reduced_intensity1d):
    pdfgen = PDFGenerator1D(reduced_intensity1d)
    pdf3 = pdfgen.get_pdf(s_min=0, s_max=15)
コード例 #4
0
def test_s_limits(reduced_intensity1d):
    pdfgen = PDFGenerator1D(reduced_intensity1d)
    pdf = pdfgen.get_pdf(s_min=0)
    pdf2 = pdfgen.get_pdf(s_min=0, s_max=10)
    assert np.array_equal(pdf.data, pdf2.data)
コード例 #5
0
def test_get_pdf(reduced_intensity1d):
    pdfgen = PDFGenerator1D(reduced_intensity1d)
    pdf = pdfgen.get_pdf(s_min=0, s_max=9)
    assert isinstance(pdf, PairDistributionFunction1D)
コード例 #6
0
def test_pdf_gen_init(reduced_intensity1d):
    pdfgen = PDFGenerator1D(reduced_intensity1d)
    assert isinstance(pdfgen, PDFGenerator1D)