1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
import pytest
import itertools
from ase.cell import Cell
def all_pbcs():
values = [False, True]
yield from itertools.product(values, values, values)
@pytest.mark.parametrize('cell', [Cell.new([3, 4, 5]), Cell.new([2, 0, 3])])
def test_uncomplete(cell):
for pbc in all_pbcs():
ucell = cell.uncomplete(pbc)
assert ucell.rank == sum(pbc & cell.any(1))
assert all(cell.uncomplete(True).any(1) == cell.any(1)), (cell.uncomplete(True), cell)
assert all(cell.uncomplete(1).any(1) == cell.any(1))
assert cell.uncomplete(False).rank == 0
assert cell.uncomplete(0).rank == 0
|
