Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
kavigupta committed May 26, 2024
1 parent d82c1b9 commit 68a5b7b
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions tests/test_out_file_cache.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@
import sys
import tempfile
import unittest
from abc import ABC, abstractmethod

from parameterized import parameterized_class

Expand All @@ -12,10 +11,9 @@
seeds = [(seed,) for seed in range(10)] if sys.platform != "win32" else []


class GenericOutFileCache(unittest.TestCase, ABC):
@abstractmethod
class GenericOutFileCacheTest(unittest.TestCase):
def get_function(self):
pass
raise NotImplementedError

def setUp(self):
# we clean this up in tearDown
Expand Down Expand Up @@ -57,7 +55,7 @@ def single_output(x, y=2, *, out_file):


@parameterized_class(("seed",), seeds)
class SingleOutputTest(GenericOutFileCache):
class SingleOutputTest(GenericOutFileCacheTest):
def get_function(self):
return cache.permacache("func", out_file="out_file")(single_output)

Expand Down Expand Up @@ -154,7 +152,7 @@ def multi_output(x, y=2, *, out_file1, out_file2):


@parameterized_class(("seed",), seeds)
class MultiOutputTest(GenericOutFileCache):
class MultiOutputTest(GenericOutFileCacheTest):
def get_function(self):
return cache.permacache("func", out_file=["out_file1", "out_file2"])(
multi_output
Expand Down

0 comments on commit 68a5b7b

Please sign in to comment.