import unittest
from src.data_structures.mockdata import MockData


class TestMockData (unittest.TestCase):

    def setUp(self):

        self.data = MockData()

    def test_random_data(self):
        data = MockData()
        a_set = data.get_random_elements(10)
        self.assertTrue(len(a_set) == 10, "the data should have 10 elements!")

    if __name__ == '__main__':
        unittest.main()