Added TestDataGenerator to generate test data. Currently generates a
list of contexts.