A python library for problems in special relativity