Python MIBDataSet._filenamesの例

コード例 #1

ファイル: test_mib.py プロジェクト: bryanfleming99/TEMproject

def test_not_too_many_files(lt_ctx):
    ds = MIBDataSet(path=MIB_TESTDATA_PATH, scan_size=(32, 32))

    with mock.patch(
            'libertem.io.dataset.mib.glob',
            side_effect=lambda p: ["/a/%d.mib" % i for i in range(256)]):
        with pytest.warns(None) as record:
            ds._filenames()

    assert len(record) == 0

コード例 #2

ファイル: test_mib.py プロジェクト: bryanfleming99/TEMproject

def test_too_many_files(lt_ctx):
    ds = MIBDataSet(path=MIB_TESTDATA_PATH, scan_size=(32, 32))

    with mock.patch(
            'libertem.io.dataset.mib.glob',
            side_effect=lambda p: ["/a/%d.mib" % i for i in range(256 * 256)]):
        with pytest.warns(RuntimeWarning) as record:
            ds._filenames()

    assert len(record) == 1
    assert "Saving data in many small files" in record[0].message.args[0]