[tf.data] refactor serialization tests based on CheckpointTestBase in kernel_tests

[kvignesh1420]

This PR is a pre-requisite for #47314 where serializability tests are being cleaned up and moved to kernel_tests.

• Renamed DatasetSerializationTestBase to CheckpointTestBase to be consistent with the future test cases.
• Refactored all serialization tests to use the new checkpoint_test_base target.
• modified targets in BUILD files.

cc: @jsimsa

[kvignesh1420]

@jsimsa I tried to run the serialization tests once before refactoring them with checkpoint_test_base.

When I run the following,

$ bazel test --runs_per_test=1 //tensorflow/python/data/experimental/kernel_tests/serialization:auto_shard_dataset_serialization_test

I get

WARNING: All specified test targets were excluded by filters
INFO: Analyzed 0 targets (0 packages loaded, 0 targets configured).
INFO: Found 0 test targets...
INFO: Elapsed time: 0.276s, Critical Path: 0.00s
INFO: 1 process: 1 internal.
INFO: Build completed successfully, 1 total action
INFO: Build completed successfully, 1 total action

Is there any issue with the current test targets or am I missing something?

UPDATE: The tests are being filtered out even after the changes.

[jsimsa]

You see the tests not being executed because they are all marked as "no_oss" (e.g. https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/data/experimental/kernel_tests/serialization/BUILD#L39). If you remove the tag, the test should execute.

[rthadur]

@kvignesh1420 can you please check sanity build failures ?

[kvignesh1420]

@rthadur I have fixed the ci sanity issues.

[kvignesh1420]

@jsimsa any update on this PR? Anything I can help with? Will continue with #47314 once this gets merged.

[jsimsa]

@kvignesh1420 there are internal projects that import the serialization module this PR moves and I will need to address the conflicts before your PR is merged, I hope to take care of that today

[kvignesh1420]

Thanks @jsimsa