Add Unit Tests
Rishit-dagli opened this issue · comments
The tests should check for the rank and shape of the output tensors, the test should override tf.test.TestCase
base class.
Feel free to take inspiration from:
- https://github.com/Rishit-dagli/Fast-Transformer/blob/main/fast_transformer/test_fast_transformer.py
- For parametrization feel free to follow https://stackoverflow.com/a/34094/11878567, can be used in the exact same way with
subTest
in TensorFlow
Hey, I'll give it a try! Thanks
I have assigned this issue to you, feel free to go over 2 of the reference links if you are new to unit testing or tf.test
in particular. Apart from the PR reviews feel free to ask if you have any more questions. Also, I'll leave it up to you if you want to address each of the checkboxes in their own PRs or combine them in a single PR.
Hey, @kaushik-42 I will take this up for now after the DM you sent me.