import gudhi

st = gudhi.SimplexTree()
# a = 0
# b = 1
# c = 2
# d = 3
# e = 4
# f = 5
# g = 6
# h = 7
# i = 8
st.insert([0,1,5],filtration=0.0)
st.insert([0,3,5],filtration=0.0)
st.insert([1,2,5],filtration=0.0)
st.insert([2,5,6],filtration=0.0)
st.insert([0,2,6],filtration=0.0)
st.insert([0,4,6],filtration=0.0)

st.insert([3,4,5],filtration=0.0)
st.insert([4,5,7],filtration=0.0)
st.insert([5,6,7],filtration=0.0)
st.insert([6,7,8],filtration=0.0)
st.insert([4,6,8],filtration=0.0)
st.insert([3,4,8],filtration=0.0)

st.insert([0,4,7],filtration=0.0)
st.insert([0,1,7],filtration=0.0)
st.insert([1,7,8],filtration=0.0)
st.insert([1,2,8],filtration=0.0)
st.insert([0,2,8],filtration=0.0)
st.insert([0,3,8],filtration=0.0)


print("number of simplices: ", st.num_simplices())

diag_Z2 =st.persistence(homology_coeff_field=2,min_persistence=2)
print("Betti numbers for Z2:", diag_Z2 )


diag_Z3 =st.persistence(homology_coeff_field=3,min_persistence=1)
print("Betti numbers for Z3:", diag_Z3 )