test:

- adds basic sanity test for ETM class;
- adds lint script

Author: Luiz Matos

conftest.py

@@ -0,0 +1,4 @@
+#!/bin/bash
+
+flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
+flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics

lint.sh

@@ -0,0 +1,18 @@
+import pytest
+
+from src import etm
+
+class TestETM:
+    def test_etm_training_with_preprocessed_reddit_dataset(self):
+        etm_instance = etm.ETM(
+            dataset='reddit-test',
+            data_path='src/datasets_for_training/min_df_0.01',
+            emb_path='src/datasets_for_training/etm_w2v_embedding.txt',
+            num_topics=5,
+            epochs=100,
+            train_embeddings=False,
+        )
+
+        etm_instance.fit(None)
+
+        assert 1 == 1