def test_connected_sublevel_sets():
array_cells = np.array([[3, 3], [2, 2], [4, 4]])
linear_cells = [3, 3, 2, 2, 4, 4]
dimensions = [2, 3]
periodic_dimensions = [False, False]
# with a numpy array version
cub = CubicalComplex(top_dimensional_cells=array_cells)
assert cub.persistence() == [(0, (2.0, float("inf")))]
assert cub.betti_numbers() == [1, 0, 0]
# with vector of dimensions
cub = CubicalComplex(dimensions=dimensions,
top_dimensional_cells=linear_cells)
assert cub.persistence() == [(0, (2.0, float("inf")))]
assert cub.betti_numbers() == [1, 0, 0]
# periodic with a numpy array version
cub = PeriodicCubicalComplex(top_dimensional_cells=array_cells,
periodic_dimensions=periodic_dimensions)
assert cub.persistence() == [(0, (2.0, float("inf")))]
assert cub.betti_numbers() == [1, 0, 0]
# periodic with vector of dimensions
cub = PeriodicCubicalComplex(dimensions=dimensions,
top_dimensional_cells=linear_cells,
periodic_dimensions=periodic_dimensions)
assert cub.persistence() == [(0, (2.0, float("inf")))]
assert cub.betti_numbers() == [1, 0, 0]
def simple_constructor(cub):
cub = CubicalComplex(dimensions=[3, 3],
top_dimensional_cells=[1, 2, 3, 4, 5, 6, 7, 8, 9])
assert cub.__is_defined() == True
assert cub.__is_persistence_defined() == False
assert cub.persistence() == [(0, (1.0, float("inf")))]
assert cub.__is_persistence_defined() == True
assert cub.betti_numbers() == [1, 0, 0]
assert cub.persistent_betti_numbers(0, 1000) == [0, 0, 0]
def test_dimension_file_constructor():
# Create test file
test_file = open("CubicalOneSphere.txt", "w")
test_file.write("2\n3\n3\n0\n0\n0\n0\n100\n0\n0\n0\n0\n")
test_file.close()
cub = CubicalComplex(perseus_file="CubicalOneSphere.txt")
assert cub.__is_defined() == True
assert cub.__is_persistence_defined() == False
assert cub.persistence() == [(1, (0.0, 100.0)), (0, (0.0, float("inf")))]
assert cub.__is_persistence_defined() == True
assert cub.betti_numbers() == [1, 0, 0]
assert cub.persistent_betti_numbers(0, 1000) == [1, 0, 0]