implement tests for generate_datasets.py and fix bugs found by the tests
implement relationship data and functions inside neo core classes