add pytest

Author: khuyentran1401

data_science_tools/pytest/get_started/sentiment.py

@@ -0,0 +1,26 @@
+from textblob import TextBlob
+
+def extract_sentiment(text: str):
+        '''Extract sentiment using textblob. 
+        Polarity is within range [-1, 1]'''
+
+        text = TextBlob(text)
+
+        return text.sentiment.polarity
+
+def test_extract_sentiment():
+
+    text = "I think today will be a great day"
+
+    sentiment = extract_sentiment(text)
+
+    assert sentiment > 0
+
+def test_extract_sentiment_negative():
+
+    text = "I do not think this will turn out well"
+
+    sentiment = extract_sentiment(text)
+
+    assert sentiment < 0
+ 

data_science_tools/pytest/parametrization/process.py

@@ -0,0 +1,18 @@
+import pytest
+def text_contain_word(word: str, text: str):
+    '''Find whether the text contains a particular word'''
+    
+    return word in text
+
+testdata = [
+    ('There is a duck in this text',True),
+    ('There is nothing here', False)
+    ]
+
+@pytest.mark.parametrize('sample, expected_output', testdata)
+def test_text_contain_word(sample, expected_output):
+
+    word = 'duck'
+
+    assert text_contain_word(word, sample) == expected_output
+

data_science_tools/pytest/parametrization/process_fixture.py

@@ -0,0 +1,33 @@
+import pytest
+from textblob import TextBlob
+
+
+def extract_sentiment(text: str):
+        '''Extract sentiment using textblob. 
+        Polarity is within range [-1, 1]'''
+
+        text = TextBlob(text)
+
+        return text.sentiment.polarity
+
+def text_contain_word(word: str, text: str):
+    '''Find whether the text contains a particular word'''
+    
+    return word in text
+
+@pytest.fixture
+def example_data():
+    return 'Today I found a duck and I am happy'
+
+
+def test_extract_sentiment(example_data):
+
+    sentiment = extract_sentiment(example_data)
+
+    assert sentiment > 0
+
+def test_text_contain_word(example_data):
+
+    word = 'duck'
+
+    assert text_contain_word(word, example_data) == True

data_science_tools/pytest/parametrization/sentiment.py

@@ -0,0 +1,19 @@
+from textblob import TextBlob
+import pytest
+
+def extract_sentiment(text: str):
+        '''Extract sentiment using textblob. 
+        Polarity is within range [-1, 1]'''
+
+        text = TextBlob(text)
+
+        return text.sentiment.polarity
+
+testdata = ["I think today will be a great day","I do not think this will turn out well"]
+
+@pytest.mark.parametrize('sample', testdata)
+def test_extract_sentiment(sample):
+
+    sentiment = extract_sentiment(sample)
+
+    assert sentiment > 0

data_science_tools/pytest/test_structure_example/src/process.py

@@ -0,0 +1,10 @@
+from textblob import TextBlob
+
+def extract_sentiment(text: str):
+        '''Extract sentiment using textblob. 
+        Polarity is within range [-1, 1]'''
+
+        text = TextBlob(text)
+
+        return text.sentiment.polarity
+        

data_science_tools/pytest/test_structure_example/tests/test_process.py

@@ -0,0 +1,15 @@
+import sys
+import os.path
+sys.path.append(
+    os.path.abspath(os.path.join(os.path.dirname(__file__), os.path.pardir)))
+from src.process import extract_sentiment
+import pytest
+
+
+def test_extract_sentiment():
+
+    text = 'Today I found a duck and I am happy'
+
+    sentiment = extract_sentiment(text)
+
+    assert sentiment > 0