-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[ci skip] Add a script to create test cases
- Loading branch information
Showing
1 changed file
with
148 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,148 @@ | ||
#!/usr/bin/env python | ||
# encoding: utf-8 | ||
|
||
""" | ||
.. codeauthor:: Tsuyoshi Hombashi <tsuyoshi.hombashi@gmail.com> | ||
""" | ||
|
||
from __future__ import print_function, unicode_literals | ||
|
||
import os | ||
|
||
import six | ||
import typepy | ||
from typepy import Infinity, Integer, Nan, NullString, RealNumber, StrictLevel, String | ||
|
||
|
||
nan = float("nan") | ||
inf = float("inf") | ||
|
||
|
||
class Params(object): | ||
|
||
def __init__(self, stream): | ||
self.__stream = stream | ||
self.strict_level = None | ||
self.typeclass = None | ||
|
||
def __del__(self): | ||
if self.__stream: | ||
self.__stream.close() | ||
self.__stream = None | ||
|
||
def write_test_params(self, method, value): | ||
if isinstance(value, six.string_types): | ||
write_value = '"{:s}"'.format(value) | ||
else: | ||
write_value = value | ||
|
||
self.__stream.write('["{:s}", {:d}, {}, {}],\n'.format( | ||
method, self.strict_level, | ||
write_value, | ||
self.__exeute_method(method, value))) | ||
|
||
def __exeute_method(self, method, value): | ||
try: | ||
result = getattr(self.typeclass(value, self.strict_level), method)() | ||
if method == "validate": | ||
result = "-" | ||
except (TypeError, typepy.TypeConversionError): | ||
return '"E"' | ||
|
||
# for string tests | ||
if NullString(result, StrictLevel.MAX).is_type(): | ||
return '""' | ||
|
||
typeobj = Integer(result, 1) | ||
if typeobj.is_type(): | ||
return typeobj.convert() | ||
|
||
typeobj = RealNumber(result, 1) | ||
if typeobj.is_type(): | ||
return typeobj.convert() | ||
|
||
if String(result, 1).is_type(): | ||
return '"{}"'.format(result) | ||
|
||
if Infinity(result, 1).is_type(): | ||
return '"inf"' | ||
|
||
if Nan(result, 1).is_type(): | ||
return '"nan"' | ||
|
||
return result | ||
|
||
|
||
class TestParamWriter(object): | ||
METHOD_LIST = ( | ||
# "is_type", | ||
# "validate", | ||
"convert", | ||
"try_convert", | ||
"force_convert", | ||
) | ||
|
||
class Bool(object): | ||
VALUE_LIST = [True, "true", 1, 1.1, None] | ||
|
||
class String(object): | ||
VALUE_LIST = ["abc", "", 1, "-1", None, True, inf, nan] | ||
|
||
class Number(object): | ||
VALUE_LIST = [ | ||
1, 1.0, 1.1, "0", "1.0", "1.1", six.MAXSIZE, True, None, inf, nan, "test", "", | ||
] | ||
|
||
def __init__(self, file_path): | ||
os.makedirs(os.path.dirname(file_path), exist_ok=True) | ||
self.__writer = Params(open(file_path, "w")) | ||
|
||
def write_bool_tests(self): | ||
self.__writer.typeclass = typepy.Bool | ||
|
||
for method in self.METHOD_LIST: | ||
for strict_level in (0, 1, 2): | ||
self.__writer.strict_level = strict_level | ||
for value in self.Bool.VALUE_LIST: | ||
self.__writer.write_test_params(method, value) | ||
|
||
def write_string_tests(self): | ||
self.__writer.typeclass = typepy.String | ||
|
||
for method in self.METHOD_LIST: | ||
for strict_level in (0, 1): | ||
self.__writer.strict_level = strict_level | ||
for value in self.String.VALUE_LIST: | ||
self.__writer.write_test_params(method, value) | ||
|
||
def write_integer_tests(self): | ||
self.__writer.typeclass = typepy.Integer | ||
|
||
for method in self.METHOD_LIST: | ||
for strict_level in (0, 1, 2): | ||
self.__writer.strict_level = strict_level | ||
for value in self.Number.VALUE_LIST: | ||
self.__writer.write_test_params(method, value) | ||
|
||
def write_realnumber_tests(self): | ||
self.__writer.typeclass = typepy.RealNumber | ||
|
||
for method in self.METHOD_LIST: | ||
for strict_level in (0, 1, 2): | ||
self.__writer.strict_level = strict_level | ||
for value in self.Number.VALUE_LIST: | ||
self.__writer.write_test_params(method, value) | ||
|
||
|
||
if __name__ == "__main__": | ||
writer = TestParamWriter("test_cases/integer.txt") | ||
writer.write_integer_tests() | ||
|
||
writer = TestParamWriter("test_cases/bool.txt") | ||
writer.write_bool_tests() | ||
|
||
writer = TestParamWriter("test_cases/realnumber.txt") | ||
writer.write_realnumber_tests() | ||
|
||
writer = TestParamWriter("test_cases/string.txt") | ||
writer.write_string_tests() |