feat(#292): w_simplicity

Author: h1alexbel

sr-data/src/sr_data/steps/numerical.py

@@ -38,9 +38,11 @@ def main(repos, out):
             "branches",
             "workflows",
             "has_release_workflow",
+            "w_simplicity",
             "tests"
         ]
     ]
-    frame["has_release_workflow"] = frame["has_release_workflow"].astype(int)
+    frame["has_release_workflow"] = frame["has_release_workflow"]
+    print(frame)
     frame.to_csv(out, index=False)
     logger.info(f"Numerical dataset created in {out}")

sr-data/src/tests/resources/to-numerical.csv

@@ -1,3 +1,3 @@
-repo,branch,readme,releases,pulls,open_issues,branches,license,workflows,w_jobs,w_oss,w_steps,has_release_workflow,tests
-foo/bar,master,"",0,0,1,1,"MIT",1,2,3,4,True,0
-bar/foo,master,"",0,0,1,1,"MIT",1,2,3,4,False,0
+repo,branch,readme,releases,pulls,open_issues,branches,license,workflows,w_jobs,w_oss,w_steps,has_release_workflow,w_simplicity,tests
+foo/bar,master,"bar",0,0,1,1,"MIT",1,2,3,4,0,0.9,0
+bar/foo,master,"foo",0,0,1,1,"MIT",1,2,3,4,1,0.85,0

sr-data/src/tests/test_numerical.py

@@ -56,18 +56,19 @@ def test_creates_csv_with_numerical(self):
                         "branches",
                         "workflows",
                         "has_release_workflow",
+                        "w_simplicity",
                         "tests"
                     ]
                 ),
                 f"Frame {frame.columns} doesn't have expected columns"
             )
             self.assertEqual(
                 frame.iloc[0]["has_release_workflow"],
-                1,
-                "has_release_workflow doesn't match with expected format"
+                0,
+                "has_release_workflow doesn't match with expected"
             )
             self.assertEqual(
                 frame.iloc[1]["has_release_workflow"],
-                0,
-                "has_release_workflow doesn't match with expected format"
+                1,
+                "has_release_workflow doesn't match with expected"
             )