from read_cifar import read_cifar def test_read_cifar(): DATA, LABELS = read_cifar(r'C:\Users\hp\Desktop\BE\image-classification\data') assert DATA.shape == (60000, 3072) assert LABELS.shape == (60000,) assert DATA.dtype == 'float32' assert LABELS.dtype == 'int64'