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'