Source code for test.test_analyze
#!usr/bin/python
"""Unittest for Analyze.py.
"""
import unittest
import os
import pandas as pd
from webcrawler.analyze import graph_scatter, graph_boxplot
__author__ = "Disaiah Bennett"
__version__ = "1.0"
[docs]class TestAnalyze(unittest.TestCase):
"""Unittest for Analyze
"""
# def __init__(self, file_name=None, columns=None, records=None, frame=None):
# self.file_name = file_name
# self.columns = columns
# self.records = records
# self.frame = frame
[docs] def test_graph_scatter_001(self):
"""Testing graph_scatter.
"""
file_name = os.listdir(".")
columns = ["Address", "Year", "Size", "Price"]
records = []
frame = pd.DataFrame()
try:
for files in file_name:
if ".csv" in files:
records.append(pd.read_csv(files, header=None, names=columns))
frame = pd.concat(records)
if records:
try:
graph_scatter(frame)
except ValueError:
pass
except OSError:
pass
[docs] def test_graph_boxplot_002(self):
"""Testing graph_boxplot.
"""
file_name = os.listdir(".")
columns = ["Address", "Year", "Size", "Price"]
records = []
frame = pd.DataFrame()
try:
for files in file_name:
if ".csv" in files:
records.append(pd.read_csv(files, header=None, names=columns))
frame = pd.concat(records)
if records:
try:
graph_boxplot(frame)
except ValueError:
pass
except OSError:
pass
if __name__ == '__main__':
unittest.main()