wehub-resource-sync
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
import numpy as np
|
|
|
|
from keras.src import testing
|
|
from keras.src.datasets import cifar10
|
|
|
|
|
|
class Cifar10LoadDataTest(testing.TestCase):
|
|
def test_x_train_shape(self):
|
|
(x_train, _), _ = cifar10.load_data()
|
|
self.assertEqual(x_train.shape, (50000, 32, 32, 3))
|
|
|
|
def test_y_train_shape(self):
|
|
(_, y_train), _ = cifar10.load_data()
|
|
self.assertEqual(y_train.shape, (50000, 1))
|
|
|
|
def test_x_test_shape(self):
|
|
_, (x_test, _) = cifar10.load_data()
|
|
self.assertEqual(x_test.shape, (10000, 32, 32, 3))
|
|
|
|
def test_y_test_shape(self):
|
|
_, (_, y_test) = cifar10.load_data()
|
|
self.assertEqual(y_test.shape, (10000, 1))
|
|
|
|
def test_x_train_dtype(self):
|
|
(x_train, _), _ = cifar10.load_data()
|
|
self.assertEqual(x_train.dtype, np.uint8)
|
|
|
|
def test_y_train_dtype(self):
|
|
(_, y_train), _ = cifar10.load_data()
|
|
self.assertEqual(y_train.dtype, np.uint8)
|
|
|
|
def test_x_test_dtype(self):
|
|
_, (x_test, _) = cifar10.load_data()
|
|
self.assertEqual(x_test.dtype, np.uint8)
|
|
|
|
def test_y_test_dtype(self):
|
|
_, (_, y_test) = cifar10.load_data()
|
|
self.assertEqual(y_test.dtype, np.uint8)
|