1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
import os
import pytest
from mp_api.client import MPRester
from mp_api.client.core import BaseRester
@pytest.fixture
def rester():
rester = BaseRester()
yield rester
rester.session.close()
@pytest.fixture()
def mpr():
rester = MPRester()
yield rester
rester.session.close()
@pytest.mark.skipif(os.getenv("MP_API_KEY", None) is None, reason="No API key found.")
@pytest.mark.xfail
def test_post_fail(rester):
rester._post_resource({}, suburl="materials/find_structure")
@pytest.mark.skipif(os.getenv("MP_API_KEY", None) is None, reason="No API key found.")
def test_pagination(mpr):
mpids = mpr.materials.search(
all_fields=False, fields=["material_id"], num_chunks=2, chunk_size=1000
)
assert len(mpids) > 1000
@pytest.mark.skipif(os.getenv("MP_API_KEY", None) is None, reason="No API key found.")
def test_count(mpr):
count = mpr.materials.count(
dict(task_ids="mp-149", _all_fields=False, _fields="material_id")
)
assert count == 1
def test_available_fields(rester, mpr):
assert len(mpr.materials.available_fields) > 0
assert rester.available_fields == ["Unknown fields."]
|
