from __future__ import absolute_import, print_function

from numpy import arange
from numpy.testing import TestCase, assert_, run_module_suite

from weave import standard_array_spec


class TestArrayConverter(TestCase):

    def test_type_match_string(self):
        s = standard_array_spec.array_converter()
        assert_(not s.type_match('string'))

    def test_type_match_int(self):
        s = standard_array_spec.array_converter()
        assert_(not s.type_match(5))

    def test_type_match_array(self):
        s = standard_array_spec.array_converter()
        assert_(s.type_match(arange(4)))


if __name__ == "__main__":
    run_module_suite()
