Update read_cifar_test.py

tests/read_cifar_test.py

 from read_cifar import read_cifar
 
-def test_read_cifar():
-    data, labels = read_cifar(r'C:\Users\etulyon1\OneDrive\Desktop\Deep_Learning1\image-classification\data')
-    assert data.shape == (60000, 3072)
-    assert labels.shape == (60000,)
-    assert data.dtype == 'float32'
-    assert labels.dtype == 'int64'
+def read_cifar_batch(BATCH_PATH):
+    with open(BATCH_PATH, "rb") as f:
+        d = pickle.load(f, encoding="bytes")
+        data = d[b"data"].astype(np.float32)
+        labels = np.array([d[b"labels"]]).astype(np.int64)
+        return data, labels