[tests-] update unit tests to use packaged sample.tsv + benchmark.csv #…

…2218
saulpw · Jan 4, 2024 · b34b2e1 · b34b2e1
1 parent 1c829d2
commit b34b2e1
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -6,6 +6,7 @@ include visidata/man/visidata.1
 include visidata/ddw/input.ddw
 include visidata/ddw/regex.ddw
 include visidata/tests/sample.tsv
+include visidata/tests/benchmark.csv
 include visidata/desktop/visidata.desktop
 include visidata/experimental/noahs_tapestry/*.json
 include visidata/experimental/noahs_tapestry/*.md

diff --git a/setup.py b/setup.py
@@ -51,7 +51,7 @@
     package_data={
         "visidata.man": ["vd.1", "vd.txt"],
         "visidata.ddw": ["input.ddw", "regex.ddw"],
-        "visidata.tests": ["sample.tsv"],
+        "visidata.tests": ["sample.tsv", "benchmark.csv"],
         "visidata.desktop": ["visidata.desktop"],
         "visidata.experimenta.noahs_tapestry": [
             "clues.json",

diff --git a/visidata/features/slide.py b/visidata/features/slide.py
@@ -100,6 +100,8 @@ def moveVisibleCol(sheet, fromVisColIdx, toVisColIdx):
 
 ## tests
 
+sample_path = vd.pkg_resources_files(visidata) / 'tests/sample.tsv'
+benchmark_path = vd.pkg_resources_files(visidata) / 'tests/benchmark.csv'
 
 def make_tester(setup_vdx):
     def t(vdx, golden):
@@ -114,8 +116,8 @@ def t(vdx, golden):
     return t
 
 def test_slide_keycol_1(vd):
-    t = make_tester('''
-            open-file sample_data/sample.tsv
+    t = make_tester(f'''
+            open-file {sample_path}
             +::OrderDate key-col
             +::Region key-col
             +::Rep key-col
@@ -139,12 +141,12 @@ def test_slide_keycol_1(vd):
 
 
 def test_slide_leftmost(vd):
-    t = make_tester('''open-file sample_data/benchmark.csv''')
+    t = make_tester(f'''open-file {benchmark_path}''')
 
     t('+::Paid slide-leftmost', 'Paid Date Customer SKU Item Quantity Unit')
 
-    t = make_tester('''
-         open-file sample_data/benchmark.csv
+    t = make_tester(f'''
+         open-file {benchmark_path}
          +::Date key-col
     ''')