def test_fetch_data_df_no_headers(mock_tab, expected_cleaned_data_no_headers):
data = get_data_from_yaml('test_fetch_data.yaml')
mock_tab.sheets_svc.get().execute.return_value = data
expected = pd.DataFrame(data=expected_cleaned_data_no_headers[1],
columns=expected_cleaned_data_no_headers[0])
assert mock_tab.fetch_data(headers=False).equals(expected)
def test_fetch_data_list_no_headers(mock_tab,
expected_cleaned_data_no_headers):
data = get_data_from_yaml('test_fetch_data.yaml')
mock_tab.sheets_svc.get().execute.return_value = data
assert mock_tab.fetch_data(
fmt='list', headers=False) == expected_cleaned_data_no_headers
def test_fetch_data_dict_no_headers(mock_tab, expected_cleaned_data_no_headers):
data = get_data_from_yaml('test_fetch_data.yaml')
mock_tab.sheets_svc.get().execute.return_value = data
keys = expected_cleaned_data_no_headers[0]
values = expected_cleaned_data_no_headers[1]
expected = [dict(zip(keys, row)) for row in values]
assert mock_tab.fetch_data(fmt='dict', headers=False) == expected
class TestCal:
data = get_data_from_yaml(path='./../case_data/test_calculatorcase_data.yaml')
@pytest.mark.run(order=1)
@pytest.mark.parametrize('a,b',data,
ids=['positive_integer', 'negative_integer', 'zero', 'super_positive_integer',
'supernegative_integer',
'float'])
def test_add(self,a,b,satrt_calculator):
"""
testcase for Calculator.add
:param a:
:param b:
:return:
"""
assert satrt_calculator.add(a,b) == a + b
@pytest.mark.run(order=3)
@pytest.mark.parametrize('a,b', data,
ids=['positive_integer', 'negative_integer', 'zero', 'super_positive_integer',
'supernegative_integer',
'float'])
def test_mul(self, a, b, satrt_calculator):
"""
testcase for Calculator.mul
:param a:
:param b:
:return:
"""
assert satrt_calculator.mul(a, b) == a * b
@pytest.mark.run(order=2)
@pytest.mark.parametrize('a,b', data,
ids=['positive_integer', 'negative_integer', 'zero', 'super_positive_integer',
'supernegative_integer',
'float'])
def test_sub(self,a,b,satrt_calculator):
"""
testcase for Calculator.sub
:param a:
:param b:
:return:
"""
assert satrt_calculator.sub(a,b) == a - b
@pytest.mark.run(order=4)
@pytest.mark.parametrize('a,b', data,
ids=['positive_integer', 'negative_integer', 'zero', 'super_positive_integer',
'supernegative_integer',
'float'])
def test_div(self,a,b,satrt_calculator):
"""
testcase for Calculator.div
:param a:
:param b:
:return:
"""
assert satrt_calculator.div(a,b) == a / b
def test_process_rows_type_errors(mock_tab, expected_data):
data = get_data_from_yaml('test_process_rows_type_error.yaml')
with pytest.raises(TypeError) as err:
mock_tab._process_rows(data)
assert err.match('Mismatch exists in expected and actual data types')
def test_process_rows(mock_tab, expected_data):
data = get_data_from_yaml('test_process_rows.yaml')
assert mock_tab._process_rows(data) == expected_data
