mirror of
https://github.com/github/semantic.git
synced 2024-12-30 02:14:20 +03:00
64 lines
2.9 KiB
Haskell
64 lines
2.9 KiB
Haskell
module Analysis.Python.Spec (spec) where
|
|
|
|
import Data.Abstract.Environment as Env
|
|
import Data.Abstract.Evaluatable (EvalError(..))
|
|
import qualified Data.Abstract.ModuleTable as ModuleTable
|
|
import Data.Abstract.Value
|
|
import qualified Language.Python.Assignment as Python
|
|
import qualified Data.Language as Language
|
|
|
|
import SpecHelpers
|
|
|
|
|
|
spec :: Spec
|
|
spec = parallel $ do
|
|
describe "Python" $ do
|
|
it "imports" $ do
|
|
((res, heap), _) <- evaluate ["main.py", "a.py", "b/__init__.py", "b/c.py"]
|
|
case ModuleTable.lookup "main.py" <$> res of
|
|
Right (Just (Module _ (addr, env) :| [])) -> do
|
|
Env.names env `shouldContain` [ "a", "b" ]
|
|
|
|
(derefQName heap ("a" :| []) env >>= deNamespace) `shouldBe` Just ("a", ["foo"])
|
|
(derefQName heap ("b" :| []) env >>= deNamespace) `shouldBe` Just ("b", ["c"])
|
|
(derefQName heap ("b" :| ["c"]) env >>= deNamespace) `shouldBe` Just ("c", ["baz"])
|
|
other -> expectationFailure (show other)
|
|
|
|
it "imports with aliases" $ do
|
|
((res, _), _) <- evaluate ["main1.py", "a.py", "b/__init__.py", "b/c.py"]
|
|
case ModuleTable.lookup "main1.py" <$> res of
|
|
Right (Just (Module _ (addr, env) :| [])) -> Env.names env `shouldContain` [ "b", "e" ]
|
|
other -> expectationFailure (show other)
|
|
|
|
it "imports using 'from' syntax" $ do
|
|
((res, _), _) <- evaluate ["main2.py", "a.py", "b/__init__.py", "b/c.py"]
|
|
case ModuleTable.lookup "main2.py" <$> res of
|
|
Right (Just (Module _ (addr, env) :| [])) -> Env.names env `shouldContain` [ "bar", "foo" ]
|
|
other -> expectationFailure (show other)
|
|
|
|
it "imports with relative syntax" $ do
|
|
((res, heap), _) <- evaluate ["main3.py", "c/__init__.py", "c/utils.py"]
|
|
case ModuleTable.lookup "main3.py" <$> res of
|
|
Right (Just (Module _ (addr, env) :| [])) -> do
|
|
Env.names env `shouldContain` [ "utils" ]
|
|
(derefQName heap ("utils" :| []) env >>= deNamespace) `shouldBe` Just ("utils", ["to_s"])
|
|
other -> expectationFailure (show other)
|
|
|
|
it "subclasses" $ do
|
|
((res, heap), _) <- evaluate ["subclass.py"]
|
|
case ModuleTable.lookup "subclass.py" <$> res of
|
|
Right (Just (Module _ (addr, env) :| [])) -> heapLookupAll addr heap `shouldBe` Just [String "\"bar\""]
|
|
other -> expectationFailure (show other)
|
|
|
|
it "handles multiple inheritance left-to-right" $ do
|
|
((res, heap), _) <- evaluate ["multiple_inheritance.py"]
|
|
case ModuleTable.lookup "multiple_inheritance.py" <$> res of
|
|
Right (Just (Module _ (addr, env) :| [])) -> heapLookupAll addr heap `shouldBe` Just [String "\"foo!\""]
|
|
other -> expectationFailure (show other)
|
|
|
|
where
|
|
ns n = Just . Latest . Last . Just . Namespace n
|
|
fixtures = "test/fixtures/python/analysis/"
|
|
evaluate = evalPythonProject . map (fixtures <>)
|
|
evalPythonProject = testEvaluating <=< evaluateProject (Proxy :: Proxy 'Language.Python) pythonParser Language.Python
|