[Task] Add LSTM and BiLSTM unit tests for session-based recommendation

[rnyak]

Description

Currently we do not have any unit test for using LSTM or BiLSTM for session-based recommendation tasks. it'd be useful to add one.

For LSTM we'd use tf.keras.layers.LSTM api.
For BiLSTM shall we use tf.keras.layers.Bidirectional?