diff --git a/tests/test_api.py b/tests/test_api.py
index dd6a7b9..c60d1a1 100644
--- a/tests/test_api.py
+++ b/tests/test_api.py
@@ -224,7 +224,7 @@ def test_PackageCacheData_return_value_contract():
assert all(isinstance(pcrec, PackageCacheRecord) for pcrec in iter_records_result)
is_writable_result = pc.is_writable
- assert is_writable_result is True or is_writable_result is False
+ assert is_writable_result in {True, False, None}
first_writable_result = PackageCacheData.first_writable()
assert isinstance(first_writable_result, PackageCacheData)
@@ -265,13 +265,14 @@ def test_PrefixData_contract():
def test_PrefixData_return_value_contract():
pd = PrefixData(context.conda_prefix)
- single_prefix_rec = next(pd.iter_records())
- get_result = pd.get(PackageRecord.from_objects(single_prefix_rec))
- assert isinstance(get_result, PrefixRecord)
+ single_prefix_rec = next(pd.iter_records(), None)
+ if single_prefix_rec:
+ get_result = pd.get(PackageRecord.from_objects(single_prefix_rec))
+ assert isinstance(get_result, PrefixRecord)
- query_result = pd.query("openssl")
- assert isinstance(query_result, tuple)
- assert all(isinstance(prefix_rec, PrefixRecord) for prefix_rec in query_result)
+ query_result = pd.query("openssl")
+ assert isinstance(query_result, tuple)
+ assert all(isinstance(prefix_rec, PrefixRecord) for prefix_rec in query_result)
iter_records_result = pd.iter_records()
assert isiterable(iter_records_result)
@@ -280,7 +281,7 @@ def test_PrefixData_return_value_contract():
)
is_writable_result = pd.is_writable
- assert is_writable_result is True or is_writable_result is False
+ assert is_writable_result in {True, False, None}
reload_result = pd.reload()
assert isinstance(reload_result, PrefixData)