from keras_core.testing.test_case import TestCase