class TestDataLoader(TestCase):
def setUp(self):
user_data_path = "consumption/tests/resource/user_data.csv"
consumption_data_path = "consumption/tests/resource/consumption/"
file_paths = (user_data_path, consumption_data_path)
self.dataloader = DataLoader(file_paths=file_paths)
def test_register_consumption(self):
self.dataloader.register_user()
self.dataloader.register_consumption()
def test_load_user_data(self):
df = self.dataloader.load_user_data()
self.assertTrue(isinstance(df, pd.DataFrame))
def test_load_consumption_data(self):
df = self.dataloader.load_consumption_data()
self.assertTrue(isinstance(df, pd.DataFrame))
def test_register_user(self):
self.assertTrue(self.dataloader.register_user())
def test_register_consumption(self):
self.dataloader.register_user()
self.assertTrue(self.dataloader.register_consumption())
class TestAggregator(TestCase):
def setUp(self):
user_data_path = "consumption/tests/resource/user_data.csv"
consumption_data_path = "consumption/tests/resource/consumption/"
file_paths = (user_data_path, consumption_data_path)
self.dataloader = DataLoader(file_paths=file_paths)
self.dataloader.register_user()
self.dataloader.register_consumption()
self.aggregator = Aggregater()
def test_mean(self):
mean_yearly = self.aggregator.mean("Y")
mean_monthly = self.aggregator.mean("M")
mean_weekly = self.aggregator.mean("W")
mean_daily = self.aggregator.mean("D")
self.assertTrue(isinstance(mean_yearly, pd.DataFrame))
self.assertTrue(isinstance(mean_monthly, pd.DataFrame))
self.assertTrue(isinstance(mean_weekly, pd.DataFrame))
self.assertTrue(isinstance(mean_daily, pd.DataFrame))
def test_total(self):
total_yearly = self.aggregator.total("Y")
total_monthly = self.aggregator.total("M")
total_weekly = self.aggregator.total("W")
total_daily = self.aggregator.total("D")
self.assertTrue(isinstance(total_yearly, pd.DataFrame))
self.assertTrue(isinstance(total_monthly, pd.DataFrame))
self.assertTrue(isinstance(total_weekly, pd.DataFrame))
self.assertTrue(isinstance(total_daily, pd.DataFrame))
def setUp(self):
user_data_path = "consumption/tests/resource/user_data.csv"
consumption_data_path = "consumption/tests/resource/consumption/"
file_paths = (user_data_path, consumption_data_path)
self.dataloader = DataLoader(file_paths=file_paths)
self.dataloader.register_user()
self.dataloader.register_consumption()
self.aggregator = Aggregater()
class TestViews(TestCase):
def setUp(self):
user_data_path = "consumption/tests/resource/user_data.csv"
consumption_data_path = "consumption/tests/resource/consumption/"
self.file_paths = (user_data_path, consumption_data_path)
self.dataloader = DataLoader(file_paths=self.file_paths)
def test_summary(self):
self.dataloader.register_user()
self.dataloader.register_consumption()
rf = RequestFactory()
request = rf.get("/summary/")
respnose = summary(request)
def handle(self, *args, **options):
is_valid = True
user_validator = UserValidator(user_data_path)
consumption_files = glob.glob(consumption_data_path + "/*.csv")
for consumption_file in consumption_files:
consumption_validator = ConsumptionValidator(consumption_file)
if not consumption_validator.is_valid():
is_valid = False
if is_valid:
print(f"All data is valid.")
dataloader = DataLoader(file_paths=file_paths)
dataloader.register_user()
dataloader.register_consumption()
def setUp(self):
user_data_path = "consumption/tests/resource/user_data.csv"
consumption_data_path = "consumption/tests/resource/consumption/"
self.file_paths = (user_data_path, consumption_data_path)
self.dataloader = DataLoader(file_paths=self.file_paths)