Final update

nogibjj · Sep 22, 2024 · dc0d633 · dc0d633
1 parent cacb8bc
commit dc0d633
Show file tree

Hide file tree

Showing 7 changed files with 32 additions and 9 deletions.
diff --git a/data/correlation_matrix.png b/data/correlation_matrix.png
diff --git a/data/histogram.png b/data/histogram.png
diff --git a/data/line_graph.png b/data/line_graph.png
diff --git a/main.ipynb b/main.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [],
    "source": [

diff --git a/requirements.txt b/requirements.txt
@@ -15,4 +15,4 @@ psutil==6.0.0
 pylint==3.2.7
 pytest==8.3.3
 ruff==0.6.6
-setuptools==73.0.1
+setuptools==73.0.1
diff --git a/src/main.py b/src/main.py
@@ -1,11 +1,19 @@
 import pandas as pd
-from lib import (
-    get_summary,
-    get_histogram,
-    get_line_graph,
-    get_correlation_matrix,
-)
 
+try:
+    from lib import (
+        get_summary,
+        get_histogram,
+        get_line_graph,
+        get_correlation_matrix,
+    )
+except Exception:
+    from src.lib import (
+        get_summary,
+        get_histogram,
+        get_line_graph,
+        get_correlation_matrix,
+    )
 
 remote_path = "https://raw.githubusercontent.com/Ramil-cyber/Ramil-Individual-Project-1/refs/heads/main/data/shanghai_ranking_2024.csv"
 

diff --git a/test/test_main.py b/test/test_main.py
@@ -1 +1,16 @@
-# from src.main import
+from src.main import calculate_iqr, save_to_md
+from src.lib import get_data
+import os
+
+remote_path = remote_path = (
+    "https://raw.githubusercontent.com/Ramil-cyber/Ramil-Individual-Project-1/refs/heads/main/data/shanghai_ranking_2024.csv"
+)
+
+
+def test_main():
+    df = get_data(remote_path)
+    top_100_df = df.iloc[:100, :]
+    print(int(calculate_iqr(top_100_df, column="Award"))) == 26
+
+    save_to_md(df)
+    assert os.path.exists("summary_statistics.md")